Fix bazel build

2024-06-04 11:13:19 +00:00 · 2024-06-04 11:13:19 +00:00 · be1d58d4fa
parent cd41a4548e
commit be1d58d4fa
1 changed files with 4 additions and 0 deletions
--- 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",