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

Reduce warning noise. 

PiperOrigin-RevId: 772941142
This commit is contained in:
Biruk Mammo 2025-06-18 09:00:47 -07:00 committed by Copybara-Service
parent 343482c7ef
commit 88284387db
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
gemma/activations.h
View File

@ -19,6 +19,7 @@
#include <math.h> // sqrtf
#include <stddef.h>
#include <atomic>
#include <vector>
#include "gemma/configs.h" // ModelConfig
@ -99,8 +100,12 @@ struct AttentionActivations {
div_seq_len(static_cast<uint32_t>(seq_len)),
query_scale(ChooseQueryScale(config)) {
// Batch size can be 0 in experimental code so do not assert.
if (batch_size == 0) {
static std::atomic_flag warned = ATOMIC_FLAG_INIT;
if (!warned.test_and_set()) {
HWY_WARN("Creating mostly empty activations with a batch_size of 0.");
}
return;
}