From c56cc34ae12f401295e600f6c9d18cb287ff6ad2 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 15 May 2026 04:30:27 +0000 Subject: [PATCH] fix: LayoutBTag is now RowMajor --- .../cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 fcfc28dd..88bfd550 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 @@ -38,7 +38,7 @@ using LayoutATag = cutlass::layout::RowMajor; constexpr int AlignmentA = 32; using ElementB = cutlass::nv_float4_t; -using LayoutBTag = cutlass::layout::ColumnMajor; +using LayoutBTag = cutlass::layout::RowMajor; constexpr int AlignmentB = 32; using ElementD = cutlass::bfloat16_t;