From 43c3daf02e177de17c208516ed3ba0d047b283bd Mon Sep 17 00:00:00 2001
From: xla authors <google-ml-automation@google.com>
Date: Fri, 14 Feb 2025 23:16:41 -0800
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 727197557
---
 xla/hlo/testlib/BUILD                                 | 1 +
 xla/hlo/testlib/hlo_hardware_independent_test_base.cc | 2 ++
 xla/hlo/testlib/hlo_hardware_independent_test_base.h  | 1 +
 3 files changed, 4 insertions(+)

diff --git a/xla/hlo/testlib/BUILD b/xla/hlo/testlib/BUILD
index 46fb09b05f1c6..a1f34221afe87 100644
--- a/xla/hlo/testlib/BUILD
+++ b/xla/hlo/testlib/BUILD
@@ -66,6 +66,7 @@ cc_library(
         "//xla:shape_util",
         "//xla:util",
         "//xla:xla_data_proto_cc",
+        "//xla:xla_proto_cc",
         "//xla/hlo/ir:hlo",
         "//xla/hlo/ir:hlo_module_group",
         "//xla/hlo/pass:hlo_pass",
diff --git a/xla/hlo/testlib/hlo_hardware_independent_test_base.cc b/xla/hlo/testlib/hlo_hardware_independent_test_base.cc
index 9fef4f365e390..b49b4356d055b 100644
--- a/xla/hlo/testlib/hlo_hardware_independent_test_base.cc
+++ b/xla/hlo/testlib/hlo_hardware_independent_test_base.cc
@@ -51,6 +51,8 @@ limitations under the License.
 #include "xla/tsl/platform/statusor.h"
 #include "xla/tsl/platform/test.h"
 #include "xla/util.h"
+#include "xla/xla.pb.h"
+#include "xla/xla_data.pb.h"
 
 namespace xla {
 
diff --git a/xla/hlo/testlib/hlo_hardware_independent_test_base.h b/xla/hlo/testlib/hlo_hardware_independent_test_base.h
index 90ddb142f0c29..9fb276d16c790 100644
--- a/xla/hlo/testlib/hlo_hardware_independent_test_base.h
+++ b/xla/hlo/testlib/hlo_hardware_independent_test_base.h
@@ -45,6 +45,7 @@ limitations under the License.
 #include "xla/shape_util.h"
 #include "xla/tsl/platform/test.h"
 #include "xla/util.h"
+#include "xla/xla.pb.h"
 #include "xla/xla_data.pb.h"
 
 namespace xla {