diff --git a/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu b/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu index 000ab028..6367e297 100644 --- a/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu +++ b/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu @@ -148,6 +148,12 @@ __global__ void remap_sf_to_cutlass_kernel( } } +///////////////////////////////////////////////////////////////////////////////////////////////// +// C API +///////////////////////////////////////////////////////////////////////////////////////////////// + +extern "C" { + int cutlass_nvfp4_gemm_run( const void* A_ptr, const void* SFA_ptr, const void* B_ptr, const void* SFB_ptr,