Coverage for cuda/pathfinder/_binaries/supported_nvidia_binaries.py: 100.00%

7 statements  

« prev     ^ index     » next       coverage.py v7.16.0, created at 2026-09-03 02:41 +0000

1# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. 

2# SPDX-License-Identifier: Apache-2.0 

3import os 

4 

5# Site-packages bin directories where binaries might be found, in search order. 

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") 

11 

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 "ptxas": (_CUDA13_BIN, _CUDA_NVCC_BIN), 

17 "nvdisasm": (_CUDA13_BIN, _CUDA_NVCC_BIN), 

18 "cuobjdump": (_CUDA13_BIN, _CUDA_NVCC_BIN), 

19 "nvprune": (_CUDA_NVCC_BIN,), 

20 "fatbinary": (_CUDA13_BIN, _CUDA_NVCC_BIN), 

21 "bin2c": (_CUDA13_BIN, _CUDA_NVCC_BIN), 

22 "nvlink": (_CUDA13_BIN, _CUDA_NVCC_BIN), 

23 # Runtime/debugging tools 

24 "cuda-gdb": (_CUDA_NVCC_BIN,), 

25 "cuda-gdbserver": (_CUDA_NVCC_BIN,), 

26 "compute-sanitizer": (_CUDA13_BIN, _CUDA_NVCC_BIN), 

27 # Profiling tools 

28 "nvprof": (_CUDA_NVCC_BIN,), 

29 "nsys": (_NSIGHT_SYSTEMS_BIN,), 

30 "nsight-sys": (_NSIGHT_SYSTEMS_BIN,), 

31 "ncu": (_NSIGHT_COMPUTE_BIN,), 

32 "nsight-compute": (_NSIGHT_COMPUTE_BIN,), 

33} 

34 

35SUPPORTED_BINARIES_ALL = SUPPORTED_BINARIES = tuple(SITE_PACKAGES_BINDIRS.keys())