Automated Code Change

PiperOrigin-RevId: 726775801
openxla · Feb 14, 2025 · 1185b9e · 1185b9e
1 parent da05dc5
commit 1185b9e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/xla/python/ifrt/ir/tests/BUILD b/xla/python/ifrt/ir/tests/BUILD
@@ -32,6 +32,7 @@ xla_cc_binary(
     testonly = True,
     srcs = ["ifrt-opt.cc"],
     deps = [
+        "//xla:xla_data_proto_cc",
         "//xla/mlir_hlo:hlo_dialect_registration",
         "//xla/pjrt:pjrt_executable",
         "//xla/python/ifrt",
@@ -61,6 +62,7 @@ xla_cc_binary(
     srcs = ["ifrt-translate.cc"],
     deps = [
         "//xla/python/ifrt:serdes",
+        "//xla/python/ifrt:serdes_proto_cc",
         "//xla/python/ifrt/ir",
         "//xla/python/ifrt/ir:ifrt_ir_program",
         "//xla/python/ifrt/ir:ifrt_ir_program_serdes",  # build_cleaner: keep

diff --git a/xla/python/ifrt/ir/tests/ifrt-opt.cc b/xla/python/ifrt/ir/tests/ifrt-opt.cc
@@ -40,6 +40,7 @@ limitations under the License.
 #include "xla/python/ifrt/mock.h"
 #include "xla/python/ifrt/shape.h"
 #include "xla/python/ifrt/support/module_parsing.h"
+#include "xla/xla_data.pb.h"
 #include "tsl/platform/init_main.h"
 
 namespace xla {

diff --git a/xla/python/ifrt/ir/tests/ifrt-translate.cc b/xla/python/ifrt/ir/tests/ifrt-translate.cc
@@ -35,6 +35,7 @@ limitations under the License.
 #include "xla/python/ifrt/ir/version.h"
 #include "xla/python/ifrt/ir/vifrt_dialect.h"
 #include "xla/python/ifrt/serdes.h"
+#include "xla/python/ifrt/serdes.pb.h"
 
 namespace xla {
 namespace ifrt {