From ba60fff38aac6dc6318fe75a1697341f4437f2e7 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Fri, 12 Jul 2024 22:24:01 -0700
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 651983211
---
 third_party/xla/xla/python/ifrt_proxy/common/BUILD         | 1 +
 third_party/xla/xla/python/ifrt_proxy/common/array_util.h  | 1 +
 third_party/xla/xla/python/ifrt_proxy/common/proto_util.cc | 1 +
 third_party/xla/xla/python/ifrt_proxy/common/types.cc      | 1 +
 4 files changed, 4 insertions(+)

diff --git a/third_party/xla/xla/python/ifrt_proxy/common/BUILD b/third_party/xla/xla/python/ifrt_proxy/common/BUILD
index 99d2b52c7b5ba3..134a8505419f8d 100644
--- a/third_party/xla/xla/python/ifrt_proxy/common/BUILD
+++ b/third_party/xla/xla/python/ifrt_proxy/common/BUILD
@@ -139,6 +139,7 @@ cc_library(
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings",
+        "@com_google_absl//absl/types:span",
         "@local_tsl//tsl/platform:statusor",
     ],
 )
diff --git a/third_party/xla/xla/python/ifrt_proxy/common/array_util.h b/third_party/xla/xla/python/ifrt_proxy/common/array_util.h
index 2ba8ff7ce42567..2e575c24f3a5de 100644
--- a/third_party/xla/xla/python/ifrt_proxy/common/array_util.h
+++ b/third_party/xla/xla/python/ifrt_proxy/common/array_util.h
@@ -22,6 +22,7 @@
 
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
+#include "absl/types/span.h"
 #include "xla/python/ifrt/dtype.h"
 #include "xla/python/ifrt/shape.h"
 
diff --git a/third_party/xla/xla/python/ifrt_proxy/common/proto_util.cc b/third_party/xla/xla/python/ifrt_proxy/common/proto_util.cc
index a9d057c2139a59..6f9bb8c8850718 100644
--- a/third_party/xla/xla/python/ifrt_proxy/common/proto_util.cc
+++ b/third_party/xla/xla/python/ifrt_proxy/common/proto_util.cc
@@ -18,6 +18,7 @@
 #include <memory>
 
 #include "absl/status/status.h"
+#include "xla/python/ifrt_proxy/common/ifrt_service.pb.h"
 #include "tsl/platform/status_to_from_proto.h"
 
 namespace xla {
diff --git a/third_party/xla/xla/python/ifrt_proxy/common/types.cc b/third_party/xla/xla/python/ifrt_proxy/common/types.cc
index 05e9dff6272d73..9d222a453c58ee 100644
--- a/third_party/xla/xla/python/ifrt_proxy/common/types.cc
+++ b/third_party/xla/xla/python/ifrt_proxy/common/types.cc
@@ -26,6 +26,7 @@
 #include "absl/types/span.h"
 #include "xla/pjrt/pjrt_common.h"
 #include "xla/python/ifrt/array.h"
+#include "xla/python/ifrt_proxy/common/ifrt_service.pb.h"
 #include "xla/python/ifrt_proxy/common/types.pb.h"
 
 namespace xla {