Fix setup.py source paths

2026-05-13 23:14:05 +00:00
parent f375c80bfe
commit 1b1c3a42fe
1 changed files with 2 additions and 1 deletions
--- a/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/setup.py
+++ b/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/setup.py
@@ -53,7 +53,8 @@ setup(
        CUDAExtension(
            name="cutlass_nvfp4_gemm._C",
            sources=[
-                "cutlass_nvfp4_gemm/pytorch_binding.cpp",
+                "pytorch_binding.cpp",
+                "cutlass_nvfp4_gemm.cu",
            ],
            extra_include_paths=extra_include_paths,
            extra_cuda_cflags=extra_cuda_cflags,