From fe3b6b8d13b751432987f7ef5a457562af21d323 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 29 May 2026 19:12:26 +0000 Subject: [PATCH] test: QK+softmax T=1 first --- tests/unit/test_qk_softmax.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_qk_softmax.cu b/tests/unit/test_qk_softmax.cu index 8bd009fe..aacd2605 100644 --- a/tests/unit/test_qk_softmax.cu +++ b/tests/unit/test_qk_softmax.cu @@ -172,7 +172,7 @@ test_qk_softmax_kernel( int main() { printf("QK+Softmax Test (HD=%d, SK=%d)\n", (int)HD, (int)SK); - const int T = 4; + const int T = 1; const float SCALE = 1.0f / sqrtf((float)HD); bf16_t* h_q = (bf16_t*)calloc(T * HD, sizeof(bf16_t));