pkgs/development/python-modules/triton/0004-nvidia-allow-static-ptxas-path.patch at python-updates

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

nixpkgs / pkgs / development / python-modules / triton / 0004-nvidia-allow-static-ptxas-path.patch

at python-updates 13 lines 476 B view raw

wrap content

 1diff --git a/python/triton/knobs.py b/python/triton/knobs.py
 2index 161f739bd..047b19d69 100644
 3--- a/python/triton/knobs.py
 4+++ b/python/triton/knobs.py
 5@@ -208,6 +208,8 @@ class env_nvidia_tool(env_base[str, NvidiaTool]):
 6         else:
 7             paths = [self.default_path]
 8 
 9+        import shlex
10+        paths.extend(shlex.split("@nixpkgsExtraBinaryPaths@"))
11         for path in paths:
12             if tool := NvidiaTool.from_path(path):
13                 return tool