Coverage for cuda / pathfinder / _binaries / supported_nvidia_binaries.py: 100.00%
7 statements
« prev ^ index » next coverage.py v7.13.5, created at 2026-04-29 01:27 +0000
« prev ^ index » next coverage.py v7.13.5, created at 2026-04-29 01:27 +0000
1# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2# SPDX-License-Identifier: Apache-2.0
3import os
5# Site-packages bin directories where binaries might be found
6# Based on NVIDIA wheel layouts (same for Linux and Windows)
7_CUDA_NVCC_BIN = os.path.join("nvidia", "cuda_nvcc", "bin")
8_CUDA13_BIN = os.path.join("nvidia", "cu13", "bin")
9_NSIGHT_SYSTEMS_BIN = os.path.join("nvidia", "nsight_systems", "bin")
10_NSIGHT_COMPUTE_BIN = os.path.join("nvidia", "nsight_compute", "bin")
12# Common CUDA binary utilities available on both Linux and Windows
13SITE_PACKAGES_BINDIRS = {
14 # Core compilation tools
15 "nvcc": (_CUDA13_BIN, _CUDA_NVCC_BIN),
16 "nvdisasm": (_CUDA13_BIN, _CUDA_NVCC_BIN),
17 "cuobjdump": (_CUDA_NVCC_BIN,),
18 "nvprune": (_CUDA_NVCC_BIN,),
19 "fatbinary": (_CUDA13_BIN, _CUDA_NVCC_BIN),
20 "bin2c": (_CUDA13_BIN, _CUDA_NVCC_BIN),
21 "nvlink": (_CUDA13_BIN, _CUDA_NVCC_BIN),
22 # Runtime/debugging tools
23 "cuda-gdb": (_CUDA_NVCC_BIN,),
24 "cuda-gdbserver": (_CUDA_NVCC_BIN,),
25 "compute-sanitizer": (_CUDA13_BIN, _CUDA_NVCC_BIN),
26 # Profiling tools
27 "nvprof": (_CUDA_NVCC_BIN,),
28 "nsys": (_NSIGHT_SYSTEMS_BIN,),
29 "nsight-sys": (_NSIGHT_SYSTEMS_BIN,),
30 "ncu": (_NSIGHT_COMPUTE_BIN,),
31 "nsight-compute": (_NSIGHT_COMPUTE_BIN,),
32}
34SUPPORTED_BINARIES_ALL = SUPPORTED_BINARIES = tuple(SITE_PACKAGES_BINDIRS.keys())