From 5592d364ec5ad6be707f641702d851a8b4026832 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Thu, 16 Oct 2025 20:24:49 -0700 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 820505039 --- third_party/xla/xla/service/gpu/autotuning/BUILD | 5 +++++ .../xla/service/gpu/autotuning/autotune_cache_key_test.cc | 1 + third_party/xla/xla/service/gpu/autotuning/autotuner_util.cc | 1 + .../xla/xla/service/gpu/autotuning/autotuner_util_test.cc | 1 + .../xla/xla/service/gpu/autotuning/dot_search_space.cc | 1 + .../xla/xla/service/gpu/autotuning/gemm_fusion_autotuner.cc | 1 + 6 files changed, 10 insertions(+) diff --git a/third_party/xla/xla/service/gpu/autotuning/BUILD b/third_party/xla/xla/service/gpu/autotuning/BUILD index b05d5a0caf6..36c41b4dadc 100644 --- a/third_party/xla/xla/service/gpu/autotuning/BUILD +++ b/third_party/xla/xla/service/gpu/autotuning/BUILD @@ -202,6 +202,7 @@ cc_library( "@com_google_absl//absl/synchronization", "@com_google_absl//absl/time", "@com_google_absl//absl/types:span", + "@com_google_protobuf//:protobuf", "@llvm-project//mlir:IR", "@local_tsl//tsl/platform:path", "@local_tsl//tsl/platform:protobuf", @@ -300,6 +301,7 @@ cc_library( "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", "@com_google_absl//absl/strings:str_format", + "@com_google_protobuf//:protobuf_lite", "@llvm-project//llvm:Support", "@local_tsl//tsl/platform:protobuf", ], @@ -408,6 +410,7 @@ xla_cc_test( "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", + "@com_google_protobuf//:protobuf", "@local_tsl//tsl/platform:path", "@local_tsl//tsl/platform:protobuf", ], @@ -447,6 +450,7 @@ cc_library( "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/synchronization", "@com_google_absl//absl/time", + "@com_google_protobuf//:protobuf", "@llvm-project//llvm:Support", "@local_tsl//tsl/platform:base64", "@local_tsl//tsl/platform:path", @@ -697,6 +701,7 @@ xla_cc_test( "@com_google_absl//absl/status:status_matchers", "@com_google_absl//absl/strings", "@com_google_googletest//:gtest", + "@com_google_protobuf//:protobuf", "@local_tsl//tsl/platform:path", "@local_tsl//tsl/platform:protobuf", ], diff --git a/third_party/xla/xla/service/gpu/autotuning/autotune_cache_key_test.cc b/third_party/xla/xla/service/gpu/autotuning/autotune_cache_key_test.cc index f6fbe1c9eff..35c5669ade2 100644 --- a/third_party/xla/xla/service/gpu/autotuning/autotune_cache_key_test.cc +++ b/third_party/xla/xla/service/gpu/autotuning/autotune_cache_key_test.cc @@ -24,6 +24,7 @@ limitations under the License. #include "absl/log/check.h" #include "absl/strings/str_format.h" #include "absl/strings/string_view.h" +#include "google/protobuf/text_format.h" #include "xla/hlo/ir/hlo_clone_context.h" #include "xla/hlo/parser/hlo_parser.h" #include "xla/stream_executor/device_description.h" diff --git a/third_party/xla/xla/service/gpu/autotuning/autotuner_util.cc b/third_party/xla/xla/service/gpu/autotuning/autotuner_util.cc index aaccefd00ed..f58e95ad1bb 100644 --- a/third_party/xla/xla/service/gpu/autotuning/autotuner_util.cc +++ b/third_party/xla/xla/service/gpu/autotuning/autotuner_util.cc @@ -36,6 +36,7 @@ limitations under the License. #include "absl/time/clock.h" #include "llvm/ADT/StringRef.h" #include "llvm/Support/SHA256.h" +#include "google/protobuf/text_format.h" #include "xla/autotune_results.pb.h" #include "xla/autotuning.pb.h" #include "xla/hlo/ir/hlo_clone_context.h" diff --git a/third_party/xla/xla/service/gpu/autotuning/autotuner_util_test.cc b/third_party/xla/xla/service/gpu/autotuning/autotuner_util_test.cc index dfbb6a54509..a71910a3a31 100644 --- a/third_party/xla/xla/service/gpu/autotuning/autotuner_util_test.cc +++ b/third_party/xla/xla/service/gpu/autotuning/autotuner_util_test.cc @@ -28,6 +28,7 @@ limitations under the License. #include "absl/status/status_matchers.h" #include "absl/strings/str_cat.h" #include "absl/strings/string_view.h" +#include "google/protobuf/text_format.h" #include "xla/autotune_results.pb.h" #include "xla/autotuning.pb.h" #include "xla/hlo/ir/hlo_computation.h" diff --git a/third_party/xla/xla/service/gpu/autotuning/dot_search_space.cc b/third_party/xla/xla/service/gpu/autotuning/dot_search_space.cc index af7271a0ce9..38cfa3143d1 100644 --- a/third_party/xla/xla/service/gpu/autotuning/dot_search_space.cc +++ b/third_party/xla/xla/service/gpu/autotuning/dot_search_space.cc @@ -29,6 +29,7 @@ limitations under the License. #include "absl/log/log.h" #include "absl/strings/str_format.h" #include "llvm/ADT/STLExtras.h" +#include "google/protobuf/repeated_field.h" #include "xla/hlo/ir/hlo_instructions.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/ir/hlo_opcode.h" diff --git a/third_party/xla/xla/service/gpu/autotuning/gemm_fusion_autotuner.cc b/third_party/xla/xla/service/gpu/autotuning/gemm_fusion_autotuner.cc index cbeb05910eb..4f4c5f947c2 100644 --- a/third_party/xla/xla/service/gpu/autotuning/gemm_fusion_autotuner.cc +++ b/third_party/xla/xla/service/gpu/autotuning/gemm_fusion_autotuner.cc @@ -41,6 +41,7 @@ limitations under the License. #include "absl/synchronization/mutex.h" #include "absl/time/time.h" #include "absl/types/span.h" +#include "google/protobuf/text_format.h" #include "xla/autotune_results.pb.h" #include "xla/autotuning.pb.h" #include "xla/backends/gpu/runtime/buffer_comparator.h"