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

debug: add printf to SF remap kernel to check flat_rank and layout shape

Browse Source
This commit is contained in:
biondizzle
2026-05-14 15:24:18 +00:00
parent 8f5322ca31
commit fb390b24e2
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -117,6 +117,14 @@ __global__ void remap_sf_to_cutlass_kernel(
constexpr int flat_rank = cute::rank_v<decltype(flat)>;
// Debug: print flat_rank and first few coordinates (only for idx 0)
if (dst_idx == 0) {
printf("[remap] flat_rank=%d, MN=%d, K_sf=%d, total=%d\n", flat_rank, MN, K_sf, total);
printf("[remap] layout shape: ");
cute::print(layout_sf.shape());
printf("\n");
}
int m, k_sf_out;
if constexpr (flat_rank == 6) {