Fix PaliGemma matmul warning

jan-wassenberg · copybara-github · commit e64ed1d4961e · 2025-10-29T11:03:18.000-07:00
PiperOrigin-RevId: 825542732
diff --git 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) {

Original file line number	Diff line number	Diff line change
`@@ -404,7 +404,7 @@ class MMConfig {`
`404`	`404`	`reserved_{} {`
`405`	`405`	`HWY_DASSERT(mr == 1 \|\| mr == 2 \|\| mr == 4);`
`406`	`406`	// Some models have K which are not multiples of `kc_multiple`.
`407`		`- if (!IsOneKC(order) && (kc % kc_multiple) != 0) {`
	`407`	`+ if (kc != K && (kc % kc_multiple) != 0) {`
`408`	`408`	`HWY_WARN("kc %zu not a multiple of kc_multiple %zu", kc, kc_multiple);`
`409`	`409`	`}`
`410`	`410`	`if (nc != N && (nc % nc_multiple) != 0) {`