Add CCCL include path for CUTLASS 3.x

2026-05-13 23:18:26 +00:00
parent 12588047fd
commit d789f5e3e0
1 changed files with 5 additions and 0 deletions
--- a/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/setup.py
+++ b/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/setup.py
@@ -27,6 +27,11 @@ include_dirs = [CUTLASS_INCLUDE_DIR]
 if os.path.exists(CUTLASS_UTIL_INCLUDE):
    include_dirs.append(CUTLASS_UTIL_INCLUDE)

+# CCCL / libcu++ headers (required by CUTLASS 3.x)
+CCCL_INCLUDE = "/usr/local/cuda-13.0/targets/x86_64-linux/include/cccl"
+if os.path.exists(CCCL_INCLUDE):
+    include_dirs.append(CCCL_INCLUDE)
+
 setup(
    name="cutlass_nvfp4_gemm",
    ext_modules=[