From a16d96f0a01aad9d08ddbe2ed09d10cc4135d122 Mon Sep 17 00:00:00 2001
From: xla authors <google-ml-automation@google.com>
Date: Fri, 14 Feb 2025 03:32:59 -0800
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 726857207
---
 xla/tools/BUILD                  | 4 ++++
 xla/tools/run_hlo_module_main.cc | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/xla/tools/BUILD b/xla/tools/BUILD
index 528c64b0f5991..4d5ac95543835 100644
--- a/xla/tools/BUILD
+++ b/xla/tools/BUILD
@@ -512,6 +512,7 @@ xla_cc_binary(
     ],
     deps = [
         ":run_hlo_module_lib",
+        ":run_hlo_module_proto_cc",
         "//xla:debug_options_flags",
         "//xla/hlo/translate/mhlo_to_hlo:translate",
         "//xla/hlo/translate/stablehlo_to_hlo:translate",
@@ -521,6 +522,9 @@ xla_cc_binary(
         "//xla/service:interpreter_plugin",
         "//xla/service:platform_util",
         "//xla/tsl/util:command_line_flags",
+        "@com_google_absl//absl/log",
+        "@com_google_absl//absl/log:check",
+        "@com_google_absl//absl/status",
         "@com_google_absl//absl/strings",
         "@llvm-project//llvm:Support",
         "@tsl//tsl/platform:logging",
diff --git a/xla/tools/run_hlo_module_main.cc b/xla/tools/run_hlo_module_main.cc
index e2dc8f82e7658..782d535e73b74 100644
--- a/xla/tools/run_hlo_module_main.cc
+++ b/xla/tools/run_hlo_module_main.cc
@@ -25,6 +25,9 @@ limitations under the License.
 #include <utility>
 #include <vector>
 
+#include "absl/log/check.h"
+#include "absl/log/log.h"
+#include "absl/status/status.h"
 #include "absl/strings/str_cat.h"
 #include "llvm/Support/FileSystem.h"
 #include "llvm/Support/MemoryBuffer.h"
@@ -36,6 +39,7 @@ limitations under the License.
 #include "xla/service/hlo_runner.h"
 #include "xla/service/platform_util.h"
 #include "xla/tools/run_hlo_module.h"
+#include "xla/tools/run_hlo_module.pb.h"
 #include "xla/tsl/util/command_line_flags.h"
 #include "tsl/platform/init_main.h"
 #include "tsl/platform/logging.h"