From 33faf2633cdd51d79e20e4f3555c800a91adb1a4 Mon Sep 17 00:00:00 2001
From: xla authors <google-ml-automation@google.com>
Date: Sat, 18 Jan 2025 18:31:05 -0800
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 717111135
---
 xla/backends/cpu/runtime/xnnpack/BUILD                    | 3 +++
 xla/backends/cpu/runtime/xnnpack/xnn_dot_thunk.h          | 1 +
 xla/backends/cpu/runtime/xnnpack/xnn_dot_thunk_test.cc    | 1 +
 xla/backends/cpu/runtime/xnnpack/xnn_fusion_thunk_test.cc | 1 +
 xla/backends/cpu/runtime/xnnpack/xnn_interop.h            | 1 +
 xla/backends/cpu/runtime/xnnpack/xnn_threadpool_test.cc   | 1 +
 6 files changed, 8 insertions(+)

diff --git a/xla/backends/cpu/runtime/xnnpack/BUILD b/xla/backends/cpu/runtime/xnnpack/BUILD
index f017949a02df2..ba14fcc59e4d2 100644
--- a/xla/backends/cpu/runtime/xnnpack/BUILD
+++ b/xla/backends/cpu/runtime/xnnpack/BUILD
@@ -78,6 +78,7 @@ cc_library(
         "//xla:util",
         "@XNNPACK",
         "@com_google_absl//absl/base:core_headers",
+        "@com_google_absl//absl/log",
         "@com_google_absl//absl/status",
         "@tsl//tsl/platform:logging",
     ],
@@ -159,6 +160,7 @@ xla_cc_test(
         "//xla:executable_run_options",
         "//xla:literal_util",
         "//xla:shape_util",
+        "//xla:xla_data_proto_cc",
         "//xla/backends/cpu/runtime:buffer_allocations",
         "//xla/backends/cpu/runtime:thunk",
         "//xla/backends/cpu/runtime:thunk_testlib",
@@ -219,6 +221,7 @@ xla_cc_test(
         "//xla:executable_run_options",
         "//xla:literal_util",
         "//xla:shape_util",
+        "//xla:xla_data_proto_cc",
         "//xla/backends/cpu/runtime:buffer_allocations",
         "//xla/backends/cpu/runtime:thunk",
         "//xla/backends/cpu/runtime:thunk_testlib",
diff --git a/xla/backends/cpu/runtime/xnnpack/xnn_dot_thunk.h b/xla/backends/cpu/runtime/xnnpack/xnn_dot_thunk.h
index 6b39fa6038c01..be1b9c21a4b17 100644
--- a/xla/backends/cpu/runtime/xnnpack/xnn_dot_thunk.h
+++ b/xla/backends/cpu/runtime/xnnpack/xnn_dot_thunk.h
@@ -28,6 +28,7 @@ limitations under the License.
 #include "xla/backends/cpu/runtime/xnnpack/xnn_fusion_thunk.h"
 #include "xla/service/buffer_assignment.h"
 #include "xla/shape.h"
+#include "xla/xla_data.pb.h"
 
 namespace xla::cpu {
 
diff --git a/xla/backends/cpu/runtime/xnnpack/xnn_dot_thunk_test.cc b/xla/backends/cpu/runtime/xnnpack/xnn_dot_thunk_test.cc
index b811e2566612d..3522f4b442aa9 100644
--- a/xla/backends/cpu/runtime/xnnpack/xnn_dot_thunk_test.cc
+++ b/xla/backends/cpu/runtime/xnnpack/xnn_dot_thunk_test.cc
@@ -24,6 +24,7 @@ limitations under the License.
 #include "xla/tsl/concurrency/async_value_ref.h"
 #include "xla/tsl/platform/statusor.h"
 #include "xla/tsl/platform/test.h"
+#include "xla/xla_data.pb.h"
 
 namespace xla::cpu {
 namespace {
diff --git a/xla/backends/cpu/runtime/xnnpack/xnn_fusion_thunk_test.cc b/xla/backends/cpu/runtime/xnnpack/xnn_fusion_thunk_test.cc
index 2ee61b734ba72..1c1471c829563 100644
--- a/xla/backends/cpu/runtime/xnnpack/xnn_fusion_thunk_test.cc
+++ b/xla/backends/cpu/runtime/xnnpack/xnn_fusion_thunk_test.cc
@@ -33,6 +33,7 @@ limitations under the License.
 #include "xla/tsl/concurrency/async_value_ref.h"
 #include "xla/tsl/platform/statusor.h"
 #include "xla/tsl/platform/test.h"
+#include "xla/xla_data.pb.h"
 
 namespace xla::cpu {
 namespace {
diff --git a/xla/backends/cpu/runtime/xnnpack/xnn_interop.h b/xla/backends/cpu/runtime/xnnpack/xnn_interop.h
index 47f6aa3d29402..a16df479a0939 100644
--- a/xla/backends/cpu/runtime/xnnpack/xnn_interop.h
+++ b/xla/backends/cpu/runtime/xnnpack/xnn_interop.h
@@ -18,6 +18,7 @@ limitations under the License.
 
 #include "xnnpack.h"
 #include "absl/base/optimization.h"
+#include "absl/log/log.h"
 #include "absl/status/status.h"
 #include "xla/util.h"
 #include "tsl/platform/logging.h"
diff --git a/xla/backends/cpu/runtime/xnnpack/xnn_threadpool_test.cc b/xla/backends/cpu/runtime/xnnpack/xnn_threadpool_test.cc
index 7cdf1dd1cb91a..2fc82497e129e 100644
--- a/xla/backends/cpu/runtime/xnnpack/xnn_threadpool_test.cc
+++ b/xla/backends/cpu/runtime/xnnpack/xnn_threadpool_test.cc
@@ -20,6 +20,7 @@ limitations under the License.
 #include <limits>
 #include <vector>
 
+#include <gtest/gtest.h>
 #include "xnnpack.h"
 #include "absl/algorithm/container.h"
 #include "pthreadpool.h"