From ffed40ac5e6d7aec450bc55126f9704e181801c0 Mon Sep 17 00:00:00 2001 From: xla authors Date: Thu, 26 Dec 2024 23:25:28 -0800 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 709975511 --- xla/service/gpu/kernels/BUILD | 8 ++++++++ xla/service/gpu/kernels/custom_kernel.h | 1 + xla/service/gpu/kernels/custom_kernel_fusion.h | 3 +++ .../gpu/kernels/cutlass_gemm_custom_kernel_test.cc | 1 + xla/service/gpu/kernels/ptx_custom_kernel.cc | 1 + xla/service/gpu/kernels/ptx_custom_kernel.h | 1 + xla/service/gpu/kernels/ptx_custom_kernel_test.cc | 2 ++ xla/service/gpu/kernels/topk_custom_kernel_test.cc | 1 + xla/service/gpu/kernels/topk_kernel.cc | 1 + xla/service/gpu/kernels/topk_kernel_test.cc | 2 ++ 10 files changed, 21 insertions(+) diff --git a/xla/service/gpu/kernels/BUILD b/xla/service/gpu/kernels/BUILD index 748078648a082..471cba37efeaf 100644 --- a/xla/service/gpu/kernels/BUILD +++ b/xla/service/gpu/kernels/BUILD @@ -32,7 +32,9 @@ cc_library( "//xla/stream_executor:device_description", "@com_google_absl//absl/base:core_headers", "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/log", "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/synchronization", "@tsl//tsl/platform:logging", @@ -157,6 +159,7 @@ cc_library( "//xla/stream_executor:stream", "//xla/stream_executor:stream_executor_h", "//xla/stream_executor:typed_kernel_factory", + "@com_google_absl//absl/log", "@com_google_absl//absl/numeric:bits", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", @@ -204,6 +207,7 @@ xla_test( "@com_google_absl//absl/random", "@com_google_absl//absl/strings", "@com_google_absl//absl/time", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", @@ -309,6 +313,7 @@ xla_test( "//xla/stream_executor:stream_executor_h", "//xla/stream_executor/cuda:cuda_platform", "//xla/tsl/lib/core:status_test_util", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:path", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", @@ -524,6 +529,7 @@ cc_library( "//xla/stream_executor:launch_dim", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", ], ) @@ -541,6 +547,8 @@ xla_test( "//xla/stream_executor:stream", "//xla/stream_executor:stream_executor_h", "//xla/stream_executor/cuda:cuda_platform", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", diff --git a/xla/service/gpu/kernels/custom_kernel.h b/xla/service/gpu/kernels/custom_kernel.h index d2cb9be9aeecd..a6e6eb5b7353f 100644 --- a/xla/service/gpu/kernels/custom_kernel.h +++ b/xla/service/gpu/kernels/custom_kernel.h @@ -20,6 +20,7 @@ limitations under the License. #include #include +#include "absl/strings/string_view.h" #include "xla/stream_executor/kernel_spec.h" #include "xla/stream_executor/launch_dim.h" diff --git a/xla/service/gpu/kernels/custom_kernel_fusion.h b/xla/service/gpu/kernels/custom_kernel_fusion.h index 741e736aceec7..2b12cc4d7557c 100644 --- a/xla/service/gpu/kernels/custom_kernel_fusion.h +++ b/xla/service/gpu/kernels/custom_kernel_fusion.h @@ -22,7 +22,10 @@ limitations under the License. #include "absl/base/thread_annotations.h" #include "absl/container/flat_hash_map.h" +#include "absl/log/log.h" #include "absl/status/status.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "absl/synchronization/mutex.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/service/gpu/kernels/custom_kernel.h" diff --git a/xla/service/gpu/kernels/cutlass_gemm_custom_kernel_test.cc b/xla/service/gpu/kernels/cutlass_gemm_custom_kernel_test.cc index 7cdc9507e3e7f..7362bfa196624 100644 --- a/xla/service/gpu/kernels/cutlass_gemm_custom_kernel_test.cc +++ b/xla/service/gpu/kernels/cutlass_gemm_custom_kernel_test.cc @@ -20,6 +20,7 @@ limitations under the License. #include #include +#include #include "xla/stream_executor/device_memory.h" #include "xla/stream_executor/kernel.h" #include "xla/stream_executor/platform.h" diff --git a/xla/service/gpu/kernels/ptx_custom_kernel.cc b/xla/service/gpu/kernels/ptx_custom_kernel.cc index b1185129afc89..21e6e56b7c711 100644 --- a/xla/service/gpu/kernels/ptx_custom_kernel.cc +++ b/xla/service/gpu/kernels/ptx_custom_kernel.cc @@ -20,6 +20,7 @@ limitations under the License. #include #include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/service/gpu/kernels/custom_kernel.h" #include "xla/stream_executor/kernel.h" #include "xla/stream_executor/kernel_spec.h" diff --git a/xla/service/gpu/kernels/ptx_custom_kernel.h b/xla/service/gpu/kernels/ptx_custom_kernel.h index 2ccb21ee8da8a..d39d6ca1baae0 100644 --- a/xla/service/gpu/kernels/ptx_custom_kernel.h +++ b/xla/service/gpu/kernels/ptx_custom_kernel.h @@ -20,6 +20,7 @@ limitations under the License. #include #include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/service/gpu/kernels/custom_kernel.h" #include "xla/stream_executor/launch_dim.h" diff --git a/xla/service/gpu/kernels/ptx_custom_kernel_test.cc b/xla/service/gpu/kernels/ptx_custom_kernel_test.cc index fae33d965a4af..e6f5ca3996d16 100644 --- a/xla/service/gpu/kernels/ptx_custom_kernel_test.cc +++ b/xla/service/gpu/kernels/ptx_custom_kernel_test.cc @@ -19,6 +19,8 @@ limitations under the License. #include #include +#include +#include "absl/strings/string_view.h" #include "xla/service/gpu/kernels/custom_kernel.h" #include "xla/stream_executor/cuda/cuda_platform.h" #include "xla/stream_executor/device_memory.h" diff --git a/xla/service/gpu/kernels/topk_custom_kernel_test.cc b/xla/service/gpu/kernels/topk_custom_kernel_test.cc index 4f6f62605996a..46cac0ecfd234 100644 --- a/xla/service/gpu/kernels/topk_custom_kernel_test.cc +++ b/xla/service/gpu/kernels/topk_custom_kernel_test.cc @@ -22,6 +22,7 @@ limitations under the License. #include #include +#include #include #include "absl/random/random.h" #include "absl/strings/ascii.h" diff --git a/xla/service/gpu/kernels/topk_kernel.cc b/xla/service/gpu/kernels/topk_kernel.cc index 1595d823b41fd..7aa0c8e294b10 100644 --- a/xla/service/gpu/kernels/topk_kernel.cc +++ b/xla/service/gpu/kernels/topk_kernel.cc @@ -23,6 +23,7 @@ limitations under the License. #include #include +#include "absl/log/log.h" #include "absl/numeric/bits.h" #include "absl/status/status.h" #include "absl/status/statusor.h" diff --git a/xla/service/gpu/kernels/topk_kernel_test.cc b/xla/service/gpu/kernels/topk_kernel_test.cc index 8ce9d80af1261..db457017091b5 100644 --- a/xla/service/gpu/kernels/topk_kernel_test.cc +++ b/xla/service/gpu/kernels/topk_kernel_test.cc @@ -23,6 +23,8 @@ limitations under the License. #include #include +#include +#include #include "absl/log/check.h" #include "absl/random/random.h" #include "absl/strings/substitute.h"