From 006999063cf7dd8dd5f1780cc4e9062bc2c1dc39 Mon Sep 17 00:00:00 2001
From: Jan Wassenberg <janwas@google.com>
Date: Wed, 29 Oct 2025 11:13:20 -0700
Subject: [PATCH] Fix PaliGemma matmul warning

PiperOrigin-RevId: 825627406
---
 ops/matmul.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ops/matmul.h b/ops/matmul.h
index cc525783..fb29bcc2 100644
--- a/ops/matmul.h
+++ b/ops/matmul.h
@@ -404,7 +404,7 @@ class MMConfig {
         reserved_{} {
     HWY_DASSERT(mr == 1 || mr == 2 || mr == 4);
     // Some models have K which are not multiples of `kc_multiple`.
-    if (!IsOneKC(order) && (kc % kc_multiple) != 0) {
+    if (kc != K && (kc % kc_multiple) != 0) {
       HWY_WARN("kc %zu not a multiple of kc_multiple %zu", kc, kc_multiple);
     }
     if (nc != N && (nc % nc_multiple) != 0) {