Automated Code Change

PiperOrigin-RevId: 687727617
openxla · Oct 20, 2024 · d57cd11 · d57cd11
1 parent 7e3e155
commit d57cd11
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 1 deletion.
diff --git a/xla/hlo/testlib/BUILD b/xla/hlo/testlib/BUILD
@@ -67,6 +67,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
@@ -45,6 +45,8 @@ limitations under the License.
 #include "xla/shape.h"
 #include "xla/shape_util.h"
 #include "xla/util.h"
+#include "xla/xla.pb.h"
+#include "xla/xla_data.pb.h"
 #include "tsl/platform/errors.h"
 #include "tsl/platform/logging.h"
 #include "tsl/platform/statusor.h"

diff --git a/xla/hlo/testlib/hlo_hardware_independent_test_base.h b/xla/hlo/testlib/hlo_hardware_independent_test_base.h
@@ -44,6 +44,7 @@ limitations under the License.
 #include "xla/shape_layout.h"
 #include "xla/shape_util.h"
 #include "xla/util.h"
+#include "xla/xla.pb.h"
 #include "xla/xla_data.pb.h"
 #include "tsl/platform/test.h"
 

diff --git a/xla/hlo/testlib/verified_hlo_module.cc b/xla/hlo/testlib/verified_hlo_module.cc
@@ -22,7 +22,6 @@ limitations under the License.
 #include "xla/util.h"
 #include "tsl/platform/errors.h"
 #include "tsl/platform/logging.h"
-#include "tsl/platform/status.h"
 #include "tsl/platform/test.h"
 
 namespace xla {

diff --git a/xla/hlo/testlib/verified_hlo_module.h b/xla/hlo/testlib/verified_hlo_module.h
@@ -17,12 +17,14 @@ limitations under the License.
 
 #include <functional>
 
+#include "absl/status/status.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
 #include "xla/service/hlo_module_config.h"
 #include "xla/service/hlo_verifier.h"
 #include "xla/shape.h"
 #include "xla/types.h"
+#include "xla/util.h"
 #include "tsl/platform/status.h"
 
 namespace xla {