From c7f5e9313665c2adc6b57153651e0684df17d36b Mon Sep 17 00:00:00 2001
From: Jan Wassenberg <janwas@google.com>
Date: Mon, 10 Jun 2024 09:34:44 -0700
Subject: [PATCH] Update benchmark with internal init

PiperOrigin-RevId: 641929308
---
 BUILD.bazel        | 1 +
 gemma/benchmark.cc | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/BUILD.bazel b/BUILD.bazel
index 9f448c1..2449a5d 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -214,6 +214,7 @@ cc_binary(
         ":common",
         ":cross_entropy",
         ":gemma_lib",
+        # "//base",
         "//compression:io",
         "@hwy//:hwy",
         "@hwy//:nanobenchmark",
diff --git a/gemma/benchmark.cc b/gemma/benchmark.cc
index 7a04f1f..db4a3ef 100644
--- a/gemma/benchmark.cc
+++ b/gemma/benchmark.cc
@@ -9,6 +9,7 @@
 #include <utility>  // std::pair
 #include <vector>
 
+// Placeholder for internal header, do not modify.
 #include "compression/io.h"  // Path
 #include "gemma/cross_entropy.h"
 #include "gemma/gemma.h"
@@ -262,6 +263,10 @@ int BenchmarkTriviaQA(gcpp::Gemma& model, gcpp::InferenceArgs& args,
  2b-it-sfp.sbs --goldens_dir "../goldens"
 */
 int main(int argc, char** argv) {
+  {
+    // Placeholder for internal init, do not modify.
+  }
+
   gcpp::LoaderArgs loader(argc, argv);
   gcpp::InferenceArgs args(argc, argv);  // inference
   gcpp::AppArgs app(argc, argv);