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

14 lines
488 B
Diff

diff --git a/python/triton/knobs.py b/python/triton/knobs.py
index 30804b170..c6a3a737d 100644
--- a/python/triton/knobs.py
+++ b/python/triton/knobs.py
@@ -203,6 +203,8 @@ class env_nvidia_tool(env_base[str, NvidiaTool]):
# accessible.
self.default(),
]
+ import shlex
+ paths.extend(shlex.split("@nixpkgsExtraBinaryPaths@"))
for path in paths:
if not path or not os.access(path, os.X_OK):
continue