Automated Code Change

PiperOrigin-RevId: 692562224

Author: Google-ML-Automation

xla/python/ifrt/ir/transforms/BUILD

@@ -116,6 +116,7 @@ cc_library(
     hdrs = ["utils.h"],
     compatible_with = get_compatible_with_portable(),
     deps = [
+        "//xla:xla_data_proto_cc",
         "//xla/mlir/utils:type_util",
         "//xla/python/ifrt",
         "//xla/python/ifrt/ir",

xla/python/ifrt/ir/transforms/utils.cc

@@ -28,6 +28,7 @@ limitations under the License.
 #include "xla/python/ifrt/ir/ifrt_dialect.h"
 #include "xla/python/ifrt/ir/ifrt_ops.h"
 #include "xla/python/pjrt_ifrt/pjrt_dtype.h"
+#include "xla/xla_data.pb.h"
 
 namespace xla {
 namespace ifrt {