No public description

PiperOrigin-RevId: 843665619
2025-12-12 06:36:40 -08:00 · 2025-12-12 06:36:40 -08:00 · 506fb22be7
parent 338cd8a36e
commit 506fb22be7
2 changed files with 2 additions and 0 deletions
--- a/gemma/configs.h
+++ b/gemma/configs.h
@ -83,6 +83,7 @@ static inline bool EnumValid(LayerAttentionType type) {
 enum class AttentionImpl {
  kOld,
  kFlash,
+  kSentinel,
 };

 AttentionImpl GetAttentionImpl(const std::string& impl);
--- a/gemma/kv_cache_test.cc
+++ b/gemma/kv_cache_test.cc
@ -1,5 +1,6 @@
 #include "gemma/kv_cache.h"

+#include <cstddef>
 #include <vector>

 #include "gtest/gtest.h"