happyz
/
gemma.cpp
mirror of https://github.com/google/gemma.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix PaliGemma matmul warning 

PiperOrigin-RevId: 825627406
This commit is contained in:
Jan Wassenberg 2025-10-29 11:13:20 -07:00 committed by Copybara-Service
parent ecab0cef3a
commit 006999063c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ops/matmul.h
View File

@ -404,7 +404,7 @@ class MMConfig {
reserved_{} { reserved_{} {
HWY_DASSERT(mr == 1 || mr == 2 || mr == 4); HWY_DASSERT(mr == 1 || mr == 2 || mr == 4);
// Some models have K which are not multiples of `kc_multiple`. // 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); HWY_WARN("kc %zu not a multiple of kc_multiple %zu", kc, kc_multiple);
} }
if (nc != N && (nc % nc_multiple) != 0) { if (nc != N && (nc % nc_multiple) != 0) {