Automated Code Change

PiperOrigin-RevId: 717111135
openxla · Jan 19, 2025 · 33faf26 · 33faf26
1 parent b21aaff
commit 33faf26
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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"