From 571fc43f574fb36a6c8733b378aa91eb10dc1ebd Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sun, 24 May 2026 00:07:22 +0000 Subject: [PATCH] D1.3: Fix while loop in cotiled diag - precompute num_tmem_alloc_cols --- tests/unit/test_d1_3_cotiled.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/unit/test_d1_3_cotiled.py b/tests/unit/test_d1_3_cotiled.py index cfef2088..2e22c92e 100644 --- a/tests/unit/test_d1_3_cotiled.py +++ b/tests/unit/test_d1_3_cotiled.py @@ -82,9 +82,10 @@ class SmemPDiag: s_cols = self.qk_mma_tiler[1] o_cols = find_tmem_tensor_col_offset(tOtO) total = max(s_cols, o_cols) - self.num_tmem_alloc_cols = 1 - while self.num_tmem_alloc_cols < total: - self.num_tmem_alloc_cols *= 2 + _n = 1 + while _n < total: + _n *= 2 + self.num_tmem_alloc_cols = _n self.tOrP0_offset = 0 # SMEM-P # Build SMEM layouts