biondizzle/nvfp4-megamoe-kernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: remove static_assert in constexpr else (build fix)

Browse Source
This commit is contained in:
biondizzle
2026-05-15 21:27:27 +00:00
parent 7285331395
commit a09d8e477e
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu
View File

@@ -136,13 +136,11 @@ __global__ void remap_sf_to_cutlass_kernel(
constexpr int LayoutRank = cute::rank_v<decltype(layout_sf.shape())>;
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;