nvidia_cuda_cccl_header_include_path_verification.py

1import os
2import nvidia.cuda_cccl
3
4# The nvidia-cuda-cccl-cu12 package provides the C++ headers 
5# for Thrust, CUB, and libcu++ to be used in CUDA applications.
6
7# Get the directory where the headers are installed
8cccl_include_path = os.path.join(os.path.dirname(nvidia.cuda_cccl.__file__), "include")
9
10if os.path.exists(cccl_include_path):
11    print(f"CCCL headers found at: {cccl_include_path}")
12    print("You can now pass this path to your NVCC compiler or JIT engine.")
13    
14    # Example: List sub-libraries included
15    print("Included libraries:", os.listdir(cccl_include_path))
16else:
17    print("CCCL include directory not found. Check your installation.")