Skip to content

Commit 6a7d08c

Browse files
authored
pathfinder: extended bin search paths with nvidia/cu13/bin (#1846)
* Extended search paths to nvidia/cu13/bin * Addressed PR comments: changed location of _CUDA13_BIN + prefering newer _CUDA13_BIN path
1 parent 64b8c07 commit 6a7d08c

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

cuda_pathfinder/cuda/pathfinder/_binaries/supported_nvidia_binaries.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,23 +5,24 @@
55
# Site-packages bin directories where binaries might be found
66
# Based on NVIDIA wheel layouts (same for Linux and Windows)
77
_CUDA_NVCC_BIN = os.path.join("nvidia", "cuda_nvcc", "bin")
8+
_CUDA13_BIN = os.path.join("nvidia", "cu13", "bin")
89
_NSIGHT_SYSTEMS_BIN = os.path.join("nvidia", "nsight_systems", "bin")
910
_NSIGHT_COMPUTE_BIN = os.path.join("nvidia", "nsight_compute", "bin")
1011

1112
# Common CUDA binary utilities available on both Linux and Windows
1213
SITE_PACKAGES_BINDIRS = {
1314
# Core compilation tools
14-
"nvcc": (_CUDA_NVCC_BIN,),
15-
"nvdisasm": (_CUDA_NVCC_BIN,),
15+
"nvcc": (_CUDA13_BIN, _CUDA_NVCC_BIN),
16+
"nvdisasm": (_CUDA13_BIN, _CUDA_NVCC_BIN),
1617
"cuobjdump": (_CUDA_NVCC_BIN,),
1718
"nvprune": (_CUDA_NVCC_BIN,),
18-
"fatbinary": (_CUDA_NVCC_BIN,),
19-
"bin2c": (_CUDA_NVCC_BIN,),
20-
"nvlink": (_CUDA_NVCC_BIN,),
19+
"fatbinary": (_CUDA13_BIN, _CUDA_NVCC_BIN),
20+
"bin2c": (_CUDA13_BIN, _CUDA_NVCC_BIN),
21+
"nvlink": (_CUDA13_BIN, _CUDA_NVCC_BIN),
2122
# Runtime/debugging tools
2223
"cuda-gdb": (_CUDA_NVCC_BIN,),
2324
"cuda-gdbserver": (_CUDA_NVCC_BIN,),
24-
"compute-sanitizer": (_CUDA_NVCC_BIN,),
25+
"compute-sanitizer": (_CUDA13_BIN, _CUDA_NVCC_BIN),
2526
# Profiling tools
2627
"nvprof": (_CUDA_NVCC_BIN,),
2728
"nsys": (_NSIGHT_SYSTEMS_BIN,),

0 commit comments

Comments
 (0)