From 8f59dceb731c3e856b2274189775a865cd054db7 Mon Sep 17 00:00:00 2001
From: hacker1024 <hacker1024@users.sourceforge.net>
Date: Wed, 5 Feb 2025 09:26:46 +1100
Subject: [PATCH] python312Packages.vllm: Respect NIX_BUILD_CORES

---
 pkgs/development/python-modules/vllm/default.nix | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pkgs/development/python-modules/vllm/default.nix b/pkgs/development/python-modules/vllm/default.nix
index 2963db2c641b..39b16aab8ec1 100644
--- a/pkgs/development/python-modules/vllm/default.nix
+++ b/pkgs/development/python-modules/vllm/default.nix
@@ -367,6 +367,12 @@ buildPythonPackage rec {
       VLLM_TARGET_DEVICE = "cpu";
     };
 
+  preConfigure = ''
+    # See: https://github.com/vllm-project/vllm/blob/v0.7.1/setup.py#L75-L109
+    # There's also NVCC_THREADS but Nix/Nixpkgs doesn't really have this concept.
+    export MAX_JOBS="$NIX_BUILD_CORES"
+  '';
+
   pythonRelaxDeps = true;
 
   pythonImportsCheck = [ "vllm" ];