From 2b4436beb694a8ce730a750893042b00cf8dd62d Mon Sep 17 00:00:00 2001
From: Krzysztof Rymski <krymski@google.com>
Date: Thu, 4 Dec 2025 02:37:21 -0800
Subject: [PATCH] Internal changes

PiperOrigin-RevId: 840151004
---
 BUILD.bazel | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/BUILD.bazel b/BUILD.bazel
index 02a7996..0e85a26 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -569,6 +569,7 @@ cc_library(
         "gemma/gemma-inl.h",
     ],
     deps = [
+        ":allocator",
         ":basics",
         ":configs",
         ":gemma_args",
@@ -576,9 +577,9 @@ cc_library(
         ":mat",
         ":matmul",
         ":matmul_env",
-        ":test_util",
         ":model_store",
         ":ops",
+        ":test_util",
         ":threading",
         ":threading_context",
         ":weights",
@@ -594,6 +595,8 @@ cc_library(
         "@highway//:profiler",
         "@highway//:thread_pool",
         "@highway//hwy/contrib/sort:vqsort",
+    ] +
+    [
     ],
 )