Handle cases where k == 1 in GEMM tester.

Microkernels with k == 1 were not being tested. PiperOrigin-RevId: 689298675
google · Oct 24, 2024 · a205406 · a205406
1 parent bdfdb7a
commit a205406
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/gemm-microkernel-tester.cc b/test/gemm-microkernel-tester.cc
@@ -69,7 +69,7 @@ TEST_P(GemmTest, Test) {
             if (params.loop_bzp_.is_set) {
               tester.b_zero_point(bzp);
             }
-            for (size_t bl = params.loop_bl_.from; bl <= tester.k() / 2;
+            for (size_t bl = params.loop_bl_.from; bl <= tester.k() == 1 ? 1 : tester.k() / 2;
                bl = params.loop_bl_.next(bl)) {
                if (params.loop_bl_.is_set) {
                 // Require block size to divide (padded) column size.