diff --git a/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/setup.py b/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/setup.py index d68fd16b..90f3802a 100644 --- 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=[