From fef7e90c0a1378079b8f6d3fa5274ec5859daa73 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 23 May 2026 21:30:59 +0000 Subject: [PATCH] D5b: Fix kernel_obj reference --- tests/unit/test_fmha_v3_stage_d5b.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_fmha_v3_stage_d5b.py b/tests/unit/test_fmha_v3_stage_d5b.py index 630aaa83..8e981091 100644 --- a/tests/unit/test_fmha_v3_stage_d5b.py +++ b/tests/unit/test_fmha_v3_stage_d5b.py @@ -24,7 +24,7 @@ def run_fmha_unnorm(q, k, v, kernel_obj, compiled_kernel, stream): m = 128 # M tile hd = v.shape[1] pv_n_tile = kernel_obj.pv_n_tile - n_pv_tiles = kernel.n_pv_tiles + n_pv_tiles = kernel_obj.n_pv_tiles c_unnorm = torch.zeros(m, hd, 1, dtype=torch.bfloat16, device='cuda') lse_tensor = torch.zeros(m, 1, 1, dtype=torch.float32, device='cuda')