Skip to content

Commit 98874ff

Browse files
committed
[FEA]: Add support for pathfinder.find_nvidia_header_directory("profiler") (#1162)
Include the profiler header descriptor and install the matching toolkit extra in the CUDA 12/13 test groups so profiler headers resolve from pip, conda, and CTK layouts. Made-with: Cursor
1 parent 5064470 commit 98874ff

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,12 @@ class HeaderDescriptorSpec:
9797
header_basename="npp.h",
9898
site_packages_dirs=("nvidia/cu13/include", "nvidia/npp/include"),
9999
),
100+
HeaderDescriptorSpec(
101+
name="profiler",
102+
packaged_with="ctk",
103+
header_basename="cuda_profiler_api.h",
104+
site_packages_dirs=("nvidia/cu13/include", "nvidia/cuda_profiler_api/include"),
105+
),
100106
HeaderDescriptorSpec(
101107
name="nvcc",
102108
packaged_with="ctk",

cuda_pathfinder/pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ test = [
1919
]
2020
# Internal organization of test dependencies.
2121
cu12 = [
22-
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti]==12.*",
22+
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler]==12.*",
2323
"cuda-toolkit[cufile]==12.*; sys_platform != 'win32'",
2424
"cutensor-cu12",
2525
"nvidia-cublasmp-cu12; sys_platform != 'win32'",
@@ -32,7 +32,7 @@ cu12 = [
3232
"nvidia-nvshmem-cu12; sys_platform != 'win32'",
3333
]
3434
cu13 = [
35-
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,nvvm]==13.*",
35+
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler,nvvm]==13.*",
3636
"cuda-toolkit[cufile]==13.*; sys_platform != 'win32'",
3737
"cutensor-cu13",
3838
"nvidia-cublasmp-cu13; sys_platform != 'win32'",

0 commit comments

Comments
 (0)