From 88284387dbac87dd26a44c01fe8d36fdcd297eca Mon Sep 17 00:00:00 2001
From: Biruk Mammo <birukw@google.com>
Date: Wed, 18 Jun 2025 09:00:47 -0700
Subject: [PATCH] Reduce warning noise.

PiperOrigin-RevId: 772941142
---
 gemma/activations.h | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/gemma/activations.h b/gemma/activations.h
index 49fc6d6..9db3dee 100644
--- a/gemma/activations.h
+++ b/gemma/activations.h
@@ -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) {
-      HWY_WARN("Creating mostly empty activations with a batch_size of 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;
     }