|
5 | 5 | # Site-packages bin directories where binaries might be found |
6 | 6 | # Based on NVIDIA wheel layouts (same for Linux and Windows) |
7 | 7 | _CUDA_NVCC_BIN = os.path.join("nvidia", "cuda_nvcc", "bin") |
| 8 | +_CUDA13_BIN = os.path.join("nvidia", "cu13", "bin") |
8 | 9 | _NSIGHT_SYSTEMS_BIN = os.path.join("nvidia", "nsight_systems", "bin") |
9 | 10 | _NSIGHT_COMPUTE_BIN = os.path.join("nvidia", "nsight_compute", "bin") |
10 | 11 |
|
11 | 12 | # Common CUDA binary utilities available on both Linux and Windows |
12 | 13 | SITE_PACKAGES_BINDIRS = { |
13 | 14 | # 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), |
16 | 17 | "cuobjdump": (_CUDA_NVCC_BIN,), |
17 | 18 | "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), |
21 | 22 | # Runtime/debugging tools |
22 | 23 | "cuda-gdb": (_CUDA_NVCC_BIN,), |
23 | 24 | "cuda-gdbserver": (_CUDA_NVCC_BIN,), |
24 | | - "compute-sanitizer": (_CUDA_NVCC_BIN,), |
| 25 | + "compute-sanitizer": (_CUDA13_BIN, _CUDA_NVCC_BIN), |
25 | 26 | # Profiling tools |
26 | 27 | "nvprof": (_CUDA_NVCC_BIN,), |
27 | 28 | "nsys": (_NSIGHT_SYSTEMS_BIN,), |
|
0 commit comments