Automated Code Change

PiperOrigin-RevId: 728069637
openxla · Feb 18, 2025 · 37a4198 · 37a4198
1 parent d83e194
commit 37a4198
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/xla/hlo/pass/BUILD b/xla/hlo/pass/BUILD
@@ -55,6 +55,7 @@ xla_cc_test(
         "//xla:shape_util",
         "//xla:test_helpers",
         "//xla:util",
+        "//xla:xla_data_proto_cc",
         "//xla/hlo/ir:hlo",
         "//xla/hlo/ir:hlo_module_group",
         "//xla/hlo/parser:hlo_parser",
@@ -86,6 +87,7 @@ cc_library(
         "//xla:status_macros",
         "//xla:types",
         "//xla:util",
+        "//xla:xla_proto_cc",
         "//xla/hlo/ir:hlo",
         "//xla/service:compilation_stats",
         "//xla/service:dump",

diff --git a/xla/hlo/pass/hlo_pass_fix_test.cc b/xla/hlo/pass/hlo_pass_fix_test.cc
@@ -38,6 +38,7 @@ limitations under the License.
 #include "xla/tsl/platform/errors.h"
 #include "xla/tsl/platform/statusor.h"
 #include "xla/util.h"
+#include "xla/xla_data.pb.h"
 
 namespace xla {
 namespace {

diff --git a/xla/hlo/pass/hlo_pass_pipeline.cc b/xla/hlo/pass/hlo_pass_pipeline.cc
@@ -33,6 +33,7 @@ limitations under the License.
 #include "xla/tsl/platform/status.h"
 #include "xla/types.h"
 #include "xla/util.h"
+#include "xla/xla.pb.h"
 #include "tsl/profiler/lib/scoped_annotation.h"
 #include "tsl/profiler/lib/traceme.h"
 

diff --git a/xla/hlo/pass/hlo_pass_pipeline.h b/xla/hlo/pass/hlo_pass_pipeline.h
@@ -31,6 +31,7 @@ limitations under the License.
 #include "xla/hlo/pass/hlo_pass_interface.h"
 #include "xla/service/compilation_stats.h"
 #include "xla/types.h"
+#include "xla/xla.pb.h"
 
 namespace xla {