test: QK+softmax T=1 first

2026-05-29 19:12:26 +00:00
parent a9a87fe7b8
commit fe3b6b8d13
1 changed files with 1 additions and 1 deletions
--- 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));