fix: use mV.iterator

2026-05-23 03:25:29 +00:00
parent a66a9efd4c
commit 81378133cc
1 changed files with 1 additions and 1 deletions
--- a/tests/unit/test_d1_diag.py
+++ b/tests/unit/test_d1_diag.py
@@ -32,7 +32,7 @@ b_major = LayoutEnum.from_tensor(mK).mma_major_mode()

 pv_n_tile = min(hd, 256)
 v_fmha = cute.make_tensor(
-    v_kernel.iterator,
+    mV.iterator,
    cute.make_layout((pv_n_tile, n_kv, 1), stride=(1, pv_n_tile, pv_n_tile * n_kv)),
 )
 v_major = LayoutEnum.from_tensor(v_fmha).mma_major_mode()