From 1b1c3a42febbc51cfee97d09032de10048beea4c Mon Sep 17 00:00:00 2001 From: biondizzle Date: Wed, 13 May 2026 23:14:05 +0000 Subject: [PATCH] Fix setup.py source paths --- src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/setup.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/setup.py b/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/setup.py index d813b700..f82528e0 100644 --- 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,