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

Warning fixes (sign mismatch), switch default 

PiperOrigin-RevId: 844679375
This commit is contained in:
Jan Wassenberg 2025-12-15 02:40:45 -08:00 committed by Copybara-Service
parent 506fb22be7
commit f50550f4ce
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
gemma/configs.h
View File

@ -107,6 +107,7 @@ static inline int AttentionImplToFlags(AttentionImpl impl,
case AttentionImpl::kOld: case AttentionImpl::kOld:
return kAttentionUseOld; return kAttentionUseOld;
case AttentionImpl::kFlash: case AttentionImpl::kFlash:
default:
return 0; return 0;
} }
} }

2
gemma/gemma.cc
View File

@ -628,7 +628,7 @@ static void GenerateTWithContinuousBatching(
const SampleFunc sample_token = const SampleFunc sample_token =
ChooseSampleFunc(runtime_config, engine, env.ctx); ChooseSampleFunc(runtime_config, engine, env.ctx);
int query_inserted = 0; size_t query_inserted = 0;
while (non_eos.Any() || query_inserted < all_queries.NumQueries()) { while (non_eos.Any() || query_inserted < all_queries.NumQueries()) {
for (size_t qi = 0; qi < qbatch.Size(); ++qi) { for (size_t qi = 0; qi < qbatch.Size(); ++qi) {
// Continue if qi slot is still processing. // Continue if qi slot is still processing.

2
gemma/query.h
View File

@ -131,7 +131,7 @@ class QBatch {
HWY_DASSERT(size_ != 0); HWY_DASSERT(size_ != 0);
HWY_DASSERT(start_ + size_ <= queries_.NumQueries()); HWY_DASSERT(start_ + size_ <= queries_.NumQueries());
query_idx_.reserve(size_); query_idx_.reserve(size_);
for (int i = 0; i < size_; ++i) { for (size_t i = 0; i < size_; ++i) {
query_idx_.push_back(start_ + i); query_idx_.push_back(start_ + i);
} }
} }