diff --git a/pyproject.toml b/pyproject.toml index 44286fc9a..550a30057 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -30,7 +30,8 @@ cu12 = [ "nvidia-cuda-nvcc-cu12", # for libNVVM "nvidia-cuda-runtime-cu12", "nvidia-cuda-nvrtc-cu12", - "nvidia-nvjitlink-cu12" + "nvidia-nvjitlink-cu12", + "nvidia-cuda-cccl-cu12", ] # TODO: Use cuda-toolkit package dependencies - e.g. cuda-toolkit[curand,nvvm,nvrtc]=13.* cu13 = [ @@ -40,7 +41,8 @@ cu13 = [ "nvidia-nvvm==13.*", "nvidia-cuda-runtime==13.*", "nvidia-cuda-nvrtc==13.*", - "nvidia-nvjitlink==13.*" + "nvidia-nvjitlink==13.*", + "nvidia-cuda-cccl==13.*", ] test = [