From 5072daa3907eafa74ecc1457be542f29ab12a9b9 Mon Sep 17 00:00:00 2001 From: brandon-b-miller Date: Fri, 22 Aug 2025 06:39:16 -0700 Subject: [PATCH] add cuda-cccl --- pyproject.toml | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) 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 = [