Skip to content

Commit 68b470f

Browse files
committed
chore: privatize global vars in supported binaries module
1 parent 0cf230d commit 68b470f

1 file changed

Lines changed: 18 additions & 20 deletions

File tree

cuda_pathfinder/cuda/pathfinder/_binaries/supported_nvidia_binaries.py

Lines changed: 18 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -4,32 +4,30 @@
44

55
# Site-packages bin directories where binaries might be found
66
# 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")
1010

1111
# Common CUDA binary utilities available on both Linux and Windows
1212
SITE_PACKAGES_BINDIRS = {
1313
# 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,),
2121
# 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,),
2525
# 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,),
3131
}
3232

3333
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

Comments
 (0)