From 3dbd3c5e7f19ad1a63d544076304d38f1b207b44 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 30 May 2026 07:00:14 +0000 Subject: [PATCH] debug: test chunk 1 only --- dsv4/kernels/attention/fmha_6warp_tma_multirow_multitile.cuh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/dsv4/kernels/attention/fmha_6warp_tma_multirow_multitile.cuh b/dsv4/kernels/attention/fmha_6warp_tma_multirow_multitile.cuh index 56dbdabf..81c6d98d 100644 --- a/dsv4/kernels/attention/fmha_6warp_tma_multirow_multitile.cuh +++ b/dsv4/kernels/attention/fmha_6warp_tma_multirow_multitile.cuh @@ -131,7 +131,9 @@ fmha_6warp_tma_multirow_multitile_kernel(FmhaTmaMultiRowMultiTileParams params) int phase = 0; // reset mbarrier phase per hd_chunk // DEBUG: skip hd_chunk > 0 to test just chunk 0 - if (hd_chunk > 0) { + // DEBUG: skip hd_chunk 0 to test just chunk 1 + bool skip_chunk = (hd_chunk == 0); // test chunk 1 only + if (skip_chunk) { if (my_row_active) { for (int d = 0; d < HD_CHUNK; d++) o_head[my_row * HD + hd_chunk_start + d] = f32_to_bf16(0.0f);