Files
nixpkgs/pkgs/development/python-modules/vllm/0005-drop-intel-reqs.patch

15 lines
694 B
Diff

diff --git a/requirements/cpu.txt b/requirements/cpu.txt
index 2db6d87ee..37f816170 100644
--- a/requirements/cpu.txt
+++ b/requirements/cpu.txt
@@ -21,9 +21,6 @@ torchvision; platform_machine != "ppc64le" and platform_machine != "s390x"
torchvision==0.23.0; platform_machine == "ppc64le"
datasets # for benchmark scripts
-# Intel Extension for PyTorch, only for x86_64 CPUs
-intel-openmp==2024.2.1; platform_machine == "x86_64"
-intel_extension_for_pytorch==2.8.0; platform_machine == "x86_64"
triton==3.2.0; platform_machine == "x86_64" # Triton is required for torch 2.6+cpu, as it is imported in torch.compile.
# Use this to gather CPU info and optimize based on ARM Neoverse cores