fix: LayoutBTag is now RowMajor

2026-05-15 04:30:27 +00:00
parent 9975558c23
commit c56cc34ae1
1 changed files with 1 additions and 1 deletions
--- a/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu
+++ b/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu
@@ -38,7 +38,7 @@ using LayoutATag  = cutlass::layout::RowMajor;
 constexpr int AlignmentA  = 32;

 using ElementB    = cutlass::nv_float4_t<cutlass::float_e2m1_t>;
-using LayoutBTag  = cutlass::layout::ColumnMajor;
+using LayoutBTag  = cutlass::layout::RowMajor;
 constexpr int AlignmentB  = 32;

 using ElementD    = cutlass::bfloat16_t;