From e7a6f44ba55753cb221db59cf06e4e35bcf5ddf6 Mon Sep 17 00:00:00 2001 From: Jan Wassenberg Date: Sat, 24 Feb 2024 12:21:19 -0800 Subject: [PATCH] Sync infra PiperOrigin-RevId: 610035407 --- BUILD => BUILD.bazel | 51 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) rename BUILD => BUILD.bazel (68%) diff --git a/BUILD b/BUILD.bazel similarity index 68% rename from BUILD rename to BUILD.bazel index 18dad30..190690b 100644 --- a/BUILD +++ b/BUILD.bazel @@ -114,3 +114,54 @@ cc_binary( "//:thread_pool", ], ) + +# copybara:strip_begin +cc_binary( + name = "run_csv", + srcs = [ + "run_csv.cc", + ], + deps = [ + ":app", + ":args", + ":gemma_lib", + "//compression:compress", + # copybara:import_next_line:hwy + "//:hwy", + # copybara:import_next_line:hwy + "//:nanobenchmark", + # copybara:import_next_line:hwy + "//:profiler", + # copybara:import_next_line:hwy + "//:thread_pool", + "//third_party/riegeli/bytes:file_reader", + "//third_party/riegeli/bytes:file_writer", + "//third_party/riegeli/csv:csv_reader", + "//third_party/riegeli/csv:csv_writer", + ], +) + +gensignature( + name = "gemma_sign", + srcs = [":gemma"], +) + +cc_test( + name = "benchmarks", + size = "large", + srcs = [ + "benchmarks.cc", + ], + tags = ["notap"], + deps = [ + ":app", + ":gemma_lib", + "//third_party/benchmark", + # copybara:import_next_line:hwy + "//:hwy", + # copybara:import_next_line:hwy + "//:thread_pool", + ], +) + +# copybara:strip_end