Automated Code Change

PiperOrigin-RevId: 723445422
openxla · Feb 5, 2025 · 2464cbd · 2464cbd
1 parent 9dbd1e1
commit 2464cbd
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/xla/python/ifrt/sharding.cc b/xla/python/ifrt/sharding.cc
@@ -44,6 +44,7 @@ limitations under the License.
 #include "xla/python/ifrt/memory.h"
 #include "xla/python/ifrt/serdes.h"
 #include "xla/python/ifrt/shape.h"
+#include "xla/python/ifrt/sharding.pb.h"
 #include "xla/tsl/concurrency/ref_count.h"
 #include "xla/tsl/platform/statusor.h"
 #include "xla/util.h"

diff --git a/xla/python/ifrt/tuple.cc b/xla/python/ifrt/tuple.cc
@@ -15,9 +15,6 @@ limitations under the License.
 
 #include "xla/python/ifrt/tuple.h"
 
-#include <memory>
-#include <vector>
-
 namespace xla {
 namespace ifrt {