From 506fb22be7d52a2a08da2fadf96c0a1ae9ef995f Mon Sep 17 00:00:00 2001
From: Martin Stolle <mstoll@google.com>
Date: Fri, 12 Dec 2025 06:36:40 -0800
Subject: [PATCH] No public description

PiperOrigin-RevId: 843665619
---
 gemma/configs.h        | 1 +
 gemma/kv_cache_test.cc | 1 +
 2 files changed, 2 insertions(+)

diff --git a/gemma/configs.h b/gemma/configs.h
index 447c246..38fab43 100644
--- 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);
diff --git a/gemma/kv_cache_test.cc b/gemma/kv_cache_test.cc
index bc1f8bc..157b3d9 100644
--- 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"