From a95d423c1ba5de950e63d1bb37d814c82c6aa61e Mon Sep 17 00:00:00 2001
From: xla authors <google-ml-automation@google.com>
Date: Fri, 10 Jan 2025 01:03:57 -0800
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 713950725
---
 xla/service/BUILD                               | 4 ++++
 xla/service/lockable_test.cc                    | 1 +
 xla/service/map_inliner.cc                      | 2 ++
 xla/service/map_inliner_test.cc                 | 1 +
 xla/service/mapped_ptr_container_sorter_test.cc | 1 +
 5 files changed, 9 insertions(+)

diff --git a/xla/service/BUILD b/xla/service/BUILD
index 2e4ba37fb2030..7b0dd97c91226 100644
--- a/xla/service/BUILD
+++ b/xla/service/BUILD
@@ -5234,6 +5234,7 @@ cc_library(
         "//xla/hlo/pass:hlo_pass",
         "//xla/hlo/utils:hlo_query",
         "@com_google_absl//absl/container:flat_hash_set",
+        "@com_google_absl//absl/log",
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings:string_view",
@@ -5293,6 +5294,7 @@ xla_cc_test(
         "//xla/tests:hlo_test_base",
         "//xla/tests:literal_test_util",
         "//xla/tests:xla_internal_test_main",  # fixdeps: keep
+        "@com_google_googletest//:gtest",
     ],
 )
 
@@ -5709,6 +5711,7 @@ xla_cc_test(
         "//xla/tsl/lib/core:status_test_util",
         "@com_google_absl//absl/functional:bind_front",
         "@com_google_absl//absl/log",
+        "@com_google_googletest//:gtest",
     ],
 )
 
@@ -5730,6 +5733,7 @@ xla_cc_test(
     deps = [
         ":lockable",
         "@com_google_absl//absl/synchronization",
+        "@com_google_googletest//:gtest",
         "@tsl//tsl/platform:env",
         "@tsl//tsl/platform:test",
         "@tsl//tsl/platform:test_main",
diff --git a/xla/service/lockable_test.cc b/xla/service/lockable_test.cc
index 9118fb9e7276b..67bf41cef0617 100644
--- a/xla/service/lockable_test.cc
+++ b/xla/service/lockable_test.cc
@@ -20,6 +20,7 @@ limitations under the License.
 #include <string>
 #include <utility>
 
+#include <gtest/gtest.h>
 #include "absl/synchronization/blocking_counter.h"
 #include "tsl/platform/env.h"
 #include "tsl/platform/test.h"
diff --git a/xla/service/map_inliner.cc b/xla/service/map_inliner.cc
index deb7b6755f6ce..7f96c1e8aa80a 100644
--- a/xla/service/map_inliner.cc
+++ b/xla/service/map_inliner.cc
@@ -19,7 +19,9 @@ limitations under the License.
 #include <vector>
 
 #include "absl/container/flat_hash_set.h"
+#include "absl/log/log.h"
 #include "absl/status/status.h"
+#include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "absl/types/span.h"
 #include "xla/hlo/ir/dfs_hlo_visitor_with_default.h"
diff --git a/xla/service/map_inliner_test.cc b/xla/service/map_inliner_test.cc
index de1511e2a6ff4..c9387108a19fa 100644
--- a/xla/service/map_inliner_test.cc
+++ b/xla/service/map_inliner_test.cc
@@ -18,6 +18,7 @@ limitations under the License.
 #include <memory>
 #include <utility>
 
+#include <gtest/gtest.h>
 #include "xla/hlo/ir/hlo_computation.h"
 #include "xla/hlo/ir/hlo_instruction.h"
 #include "xla/hlo/ir/hlo_opcode.h"
diff --git a/xla/service/mapped_ptr_container_sorter_test.cc b/xla/service/mapped_ptr_container_sorter_test.cc
index ca738619aa8ab..bb1b55ccdd646 100644
--- a/xla/service/mapped_ptr_container_sorter_test.cc
+++ b/xla/service/mapped_ptr_container_sorter_test.cc
@@ -22,6 +22,7 @@ limitations under the License.
 #include <utility>
 #include <vector>
 
+#include <gtest/gtest.h>
 #include "absl/functional/bind_front.h"
 #include "absl/log/log.h"
 #include "xla/test.h"