pkgs/development/python-modules/vllm/0005-drop-intel-reqs.patch at master · tjh.dev/nixpkgs

tjh.dev / nixpkgs

Clone of https://github.com/NixOS/nixpkgs.git (to stress-test knotserver)

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

at master 694 B view raw

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