debug: try all-lane MMA + print tmem_base

This commit is contained in:
2026-05-28 08:37:02 +00:00
parent d3510980e4
commit bfb1e177ce

View File

@@ -68,7 +68,14 @@ fmha_qk_verify(
uint64_t desc_k = make_umma_desc_k_none(sK_smem, HD);
// QK GEMM: S = Q @ K^T (SS, both SMEM → TMEM)
if (wid == 0 && lane == 0) {
// MMA: called by ONE lane (elect_one_sync pattern)
if (tid == 0) printf("[qk] tmem_base=%u sQ_smem=0x%x sK_smem=0x%x\n", tmem_base, sQ_smem, sK_smem);
__syncthreads();
// Try: ALL 32 lanes of warp 0 call MMA (not just lane 0)
// The CUTLASS code uses elect_one_sync which selects lane 0,
// but maybe the PTX requires all lanes to participate?
if (wid == 0) {
umma_ss_f16(tmem_base, desc_q, desc_k, false);
}
__syncwarp();