From 290ad334fc6848d632ec6f4f73b34435f22f8307 Mon Sep 17 00:00:00 2001
From: Balazs Racz <bracz@google.com>
Date: Fri, 10 Apr 2026 04:34:03 -0700
Subject: [PATCH] internal changes

PiperOrigin-RevId: 897620425
---
 gemma/tiled_attention_test.cc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/gemma/tiled_attention_test.cc b/gemma/tiled_attention_test.cc
index 79282c2d..a41c1a9e 100644
--- a/gemma/tiled_attention_test.cc
+++ b/gemma/tiled_attention_test.cc
@@ -1,13 +1,12 @@
 #include <stddef.h>
 
-#include <algorithm>
-#include <cmath>
 #include <iostream>
 #include <memory>
 #include <optional>
 #include <tuple>
 #include <vector>
 
+#include "third_party/absl/types/span.h"
 #include "compression/types.h"  // GEMMA_DISABLED_TARGETS
 #include "gemma/activations.h"
 #include "gemma/configs.h"