tjh.dev / nixpkgs
Clone of https://github.com/NixOS/nixpkgs.git (to stress-test knotserver)
fork atom
nixpkgs / pkgs / development / python-modules / triton / 0004-nvidia-allow-static-ptxas-path.patch
at gcc-offload 26 lines 901 B view raw
1From e503e572b6d444cd27f1cdf124aaf553aa3a8665 Mon Sep 17 00:00:00 2001 2From: SomeoneSerge <else+aalto@someonex.net> 3Date: Mon, 14 Oct 2024 00:12:05 +0000 4Subject: [PATCH 4/4] nvidia: allow static ptxas path 5 6--- 7 third_party/nvidia/backend/compiler.py | 3 +++ 8 1 file changed, 3 insertions(+) 9 10diff --git a/third_party/nvidia/backend/compiler.py b/third_party/nvidia/backend/compiler.py 11index 6d7994923..6720e8f97 100644 12--- a/third_party/nvidia/backend/compiler.py 13+++ b/third_party/nvidia/backend/compiler.py 14@@ -20,6 +20,9 @@ def _path_to_binary(binary: str): 15 os.path.join(os.path.dirname(__file__), "bin", binary), 16 ] 17 18+ import shlex 19+ paths.extend(shlex.split("@nixpkgsExtraBinaryPaths@")) 20+ 21 for bin in paths: 22 if os.path.exists(bin) and os.path.isfile(bin): 23 result = subprocess.check_output([bin, "--version"], stderr=subprocess.STDOUT) 24-- 252.46.0 26