From 68f745d578f5ecf18cc2036f04a34b74ff71a777 Mon Sep 17 00:00:00 2001
From: ruro <ruro.ruro@ya.ru>
Date: Wed, 17 Apr 2024 20:17:08 +0300
Subject: [PATCH] python311Packages.openai-triton: use requested number of
 cores for build

---
 pkgs/development/python-modules/openai-triton/default.nix | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pkgs/development/python-modules/openai-triton/default.nix b/pkgs/development/python-modules/openai-triton/default.nix
index 018852bdc743..2bdb8d918af3 100644
--- a/pkgs/development/python-modules/openai-triton/default.nix
+++ b/pkgs/development/python-modules/openai-triton/default.nix
@@ -115,6 +115,9 @@ buildPythonPackage rec {
 
   # Avoid GLIBCXX mismatch with other cuda-enabled python packages
   preConfigure = ''
+    # Ensure that the build process uses the requested number of cores
+    export MAX_JOBS="$NIX_BUILD_CORES"
+
     # Upstream's setup.py tries to write cache somewhere in ~/
     export HOME=$(mktemp -d)