cute.size() returns a CuTeDSL symbol, not a Python int. range() on a symbol can't iterate — the loop never unrolls. Now n_kv_tiles is computed in __init__ as s_k // 128 (Python int).
cute.size() returns a CuTeDSL symbol, not a Python int. range() on a symbol can't iterate — the loop never unrolls. Now n_kv_tiles is computed in __init__ as s_k // 128 (Python int).