|
4 | 4 |
|
5 | 5 | # Site-packages bin directories where binaries might be found |
6 | 6 | # Based on NVIDIA wheel layouts (same for Linux and Windows) |
7 | | -CUDA_NVCC_BIN = os.path.join("nvidia", "cuda_nvcc", "bin") |
8 | | -NSIGHT_SYSTEMS_BIN = os.path.join("nvidia", "nsight_systems", "bin") |
9 | | -NSIGHT_COMPUTE_BIN = os.path.join("nvidia", "nsight_compute", "bin") |
| 7 | +_CUDA_NVCC_BIN = os.path.join("nvidia", "cuda_nvcc", "bin") |
| 8 | +_NSIGHT_SYSTEMS_BIN = os.path.join("nvidia", "nsight_systems", "bin") |
| 9 | +_NSIGHT_COMPUTE_BIN = os.path.join("nvidia", "nsight_compute", "bin") |
10 | 10 |
|
11 | 11 | # Common CUDA binary utilities available on both Linux and Windows |
12 | 12 | SITE_PACKAGES_BINDIRS = { |
13 | 13 | # Core compilation tools |
14 | | - "nvcc": (CUDA_NVCC_BIN,), |
15 | | - "nvdisasm": (CUDA_NVCC_BIN,), |
16 | | - "cuobjdump": (CUDA_NVCC_BIN,), |
17 | | - "nvprune": (CUDA_NVCC_BIN,), |
18 | | - "fatbinary": (CUDA_NVCC_BIN,), |
19 | | - "bin2c": (CUDA_NVCC_BIN,), |
20 | | - "nvlink": (CUDA_NVCC_BIN,), |
| 14 | + "nvcc": (_CUDA_NVCC_BIN,), |
| 15 | + "nvdisasm": (_CUDA_NVCC_BIN,), |
| 16 | + "cuobjdump": (_CUDA_NVCC_BIN,), |
| 17 | + "nvprune": (_CUDA_NVCC_BIN,), |
| 18 | + "fatbinary": (_CUDA_NVCC_BIN,), |
| 19 | + "bin2c": (_CUDA_NVCC_BIN,), |
| 20 | + "nvlink": (_CUDA_NVCC_BIN,), |
21 | 21 | # Runtime/debugging tools |
22 | | - "cuda-gdb": (CUDA_NVCC_BIN,), |
23 | | - "cuda-gdbserver": (CUDA_NVCC_BIN,), |
24 | | - "compute-sanitizer": (CUDA_NVCC_BIN,), |
| 22 | + "cuda-gdb": (_CUDA_NVCC_BIN,), |
| 23 | + "cuda-gdbserver": (_CUDA_NVCC_BIN,), |
| 24 | + "compute-sanitizer": (_CUDA_NVCC_BIN,), |
25 | 25 | # Profiling tools |
26 | | - "nvprof": (CUDA_NVCC_BIN,), |
27 | | - "nsys": (NSIGHT_SYSTEMS_BIN,), |
28 | | - "nsight-sys": (NSIGHT_SYSTEMS_BIN,), |
29 | | - "ncu": (NSIGHT_COMPUTE_BIN,), |
30 | | - "nsight-compute": (NSIGHT_COMPUTE_BIN,), |
| 26 | + "nvprof": (_CUDA_NVCC_BIN,), |
| 27 | + "nsys": (_NSIGHT_SYSTEMS_BIN,), |
| 28 | + "nsight-sys": (_NSIGHT_SYSTEMS_BIN,), |
| 29 | + "ncu": (_NSIGHT_COMPUTE_BIN,), |
| 30 | + "nsight-compute": (_NSIGHT_COMPUTE_BIN,), |
31 | 31 | } |
32 | 32 |
|
33 | 33 | SUPPORTED_BINARIES_ALL = SUPPORTED_BINARIES = tuple(SITE_PACKAGES_BINDIRS.keys()) |
34 | | - |
35 | | -del CUDA_NVCC_BIN, NSIGHT_SYSTEMS_BIN, NSIGHT_COMPUTE_BIN, os |
0 commit comments