From 81378133cc007b2357b9f9feb262fae9d1df42ba Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Sat, 23 May 2026 03:25:29 +0000
Subject: [PATCH] fix: use mV.iterator

---
 tests/unit/test_d1_diag.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/test_d1_diag.py b/tests/unit/test_d1_diag.py
index fbc07d1b..e68f4038 100644
--- 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()