From be1d58d4fa223a3ed2df3720c0a15804cb57ff87 Mon Sep 17 00:00:00 2001
From: Zoltan Szabadka <szabadka@google.com>
Date: Tue, 4 Jun 2024 11:13:19 +0000
Subject: [PATCH] Fix bazel build

---
 BUILD.bazel | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/BUILD.bazel b/BUILD.bazel
index 59ffa69..96f9c12 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -59,8 +59,12 @@ cc_library(
         "gemma/gemma.cc",
     ],
     hdrs = [
+        "gemma/activations.h",
+        "gemma/common.h",
+        "gemma/common-inl.h",
         "gemma/configs.h",
         "gemma/gemma.h",
+        "gemma/weights.h",
     ],
     deps = [
         ":ops",