diff --git a/third_party/xla/xla/service/gpu/model/experimental/BUILD b/third_party/xla/xla/service/gpu/model/experimental/BUILD
index 535281e3433..afcbaa8ed2f 100644
--- a/third_party/xla/xla/service/gpu/model/experimental/BUILD
+++ b/third_party/xla/xla/service/gpu/model/experimental/BUILD
@@ -25,6 +25,7 @@ cc_library(
         "symbolic_tile_propagation.h",
     ],
     deps = [
+        "//xla:xla_data_proto_cc",
         "//xla/hlo/analysis:indexing_analysis",
         "//xla/hlo/ir:hlo",
         "//xla/service/gpu/model:constraint_expression",
diff --git a/third_party/xla/xla/service/gpu/model/experimental/symbolic_tile_propagation.cc b/third_party/xla/xla/service/gpu/model/experimental/symbolic_tile_propagation.cc
index 15a3d244c73..71454b5751c 100644
--- a/third_party/xla/xla/service/gpu/model/experimental/symbolic_tile_propagation.cc
+++ b/third_party/xla/xla/service/gpu/model/experimental/symbolic_tile_propagation.cc
@@ -32,6 +32,7 @@ limitations under the License.
 #include "xla/hlo/ir/hlo_opcode.h"
 #include "xla/service/gpu/model/constraint_expression.h"
 #include "xla/service/gpu/model/experimental/symbolic_tile.h"
+#include "xla/xla_data.pb.h"
 
 namespace xla::gpu {
 namespace {