15 lines
599 B
Diff
15 lines
599 B
Diff
diff --git a/third_party/nvidia/backend/compiler.py b/third_party/nvidia/backend/compiler.py
|
|
index 960334744..269e22e6e 100644
|
|
--- a/third_party/nvidia/backend/compiler.py
|
|
+++ b/third_party/nvidia/backend/compiler.py
|
|
@@ -38,6 +38,9 @@ def _path_to_binary(binary: str):
|
|
os.path.join(os.path.dirname(__file__), "bin", binary),
|
|
]
|
|
|
|
+ import shlex
|
|
+ paths.extend(shlex.split("@nixpkgsExtraBinaryPaths@"))
|
|
+
|
|
for path in paths:
|
|
if os.path.exists(path) and os.path.isfile(path):
|
|
result = subprocess.check_output([path, "--version"], stderr=subprocess.STDOUT)
|