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 af37ada1..0dc6303a 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 @@ -136,13 +136,11 @@ __global__ void remap_sf_to_cutlass_kernel( constexpr int LayoutRank = cute::rank_v; - int dst_idx; + int dst_idx = 0; if constexpr (LayoutRank == 2) { dst_idx = layout_sf(cute::make_coord(mn, k_sf * InputSFVectorSize)); } else if constexpr (LayoutRank == 3) { dst_idx = layout_sf(cute::make_coord(mn, k_sf * InputSFVectorSize, 0)); - } else { - static_assert(LayoutRank == 2 || LayoutRank == 3, "Unexpected SF layout rank"); } int src_idx = mn * src_stride_mn + k_sf * src_stride_ksf;