diff --git a/setup.py b/setup.py index 513b4355..617cd310 100644 --- a/setup.py +++ b/setup.py @@ -129,6 +129,7 @@ def get_nvcc_thread_args(): Path(this_dir) / "csrc" / "sm90", Path(this_dir) / "csrc" / "cutlass" / "include", Path(this_dir) / "csrc" / "cutlass" / "tools" / "util" / "include", + Path(CUDA_HOME) / "include" / "cccl", ], ) )