From 9198ed734f8f69cde30ba9dd08069062d9e58aca Mon Sep 17 00:00:00 2001 From: biondizzle Date: Thu, 28 May 2026 14:32:10 +0000 Subject: [PATCH] Test 1 PV K-tile from (128,128) P at offset 0 --- tests/unit/test_pv_ss_128.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/test_pv_ss_128.cu b/tests/unit/test_pv_ss_128.cu index c7ed1566..71b24b31 100644 --- a/tests/unit/test_pv_ss_128.cu +++ b/tests/unit/test_pv_ss_128.cu @@ -72,7 +72,7 @@ test_pv_ss_128() { uint64_t dv = make_umma_desc_kmajor_none(__cvta_generic_to_shared(sV), 16); uint32_t idesc = make_idesc(BLOCK_MN, HD); - for (int kt = 0; kt < 8; kt++) { + for (int kt = 0; kt < 1; kt++) { bf16_t* sp = sP + kt * 2048; uint64_t dp = make_umma_desc_kmajor_none(__cvta_generic_to_shared(sp), BLOCK_MN); if (tid == 0) umma_ss_f16(tb, dp, dv, idesc, true); // ALL accumulate @@ -97,7 +97,7 @@ test_pv_ss_128() printf("O[0,0..15] (raw): "); for (int d=0;d