From af9955e0c0de35220188fdd24d6c441a7a81a61f Mon Sep 17 00:00:00 2001
From: xla authors <google-ml-automation@google.com>
Date: Fri, 21 Mar 2025 02:26:12 -0700
Subject: [PATCH] Automated Code Change
PiperOrigin-RevId: 739105502
---
xla/backends/cpu/BUILD | 1 +
xla/backends/cpu/constant_allocation.cc | 1 +
2 files changed, 2 insertions(+)
diff --git a/xla/backends/cpu/BUILD b/xla/backends/cpu/BUILD
index c65516d819b73..726b57671f6e8 100644
--- a/xla/backends/cpu/BUILD
+++ b/xla/backends/cpu/BUILD
@@ -114,6 +114,7 @@ cc_library(
"//xla:literal",
"//xla:shape_util",
"//xla:util",
+ "//xla:xla_data_proto_cc",
"//xla/hlo/ir:hlo",
"//xla/service:buffer_assignment",
"//xla/stream_executor:device_memory",
diff --git a/xla/backends/cpu/constant_allocation.cc b/xla/backends/cpu/constant_allocation.cc
index 9da4f81481098..3702811c48e09 100644
--- a/xla/backends/cpu/constant_allocation.cc
+++ b/xla/backends/cpu/constant_allocation.cc
@@ -35,6 +35,7 @@ limitations under the License.
#include "xla/stream_executor/device_memory.h"
#include "xla/tsl/platform/statusor.h"
#include "xla/util.h"
+#include "xla/xla_data.pb.h"
namespace xla::cpu {