From 4dae6506f7e7007d5fd0c0f23bda82d41d03d803 Mon Sep 17 00:00:00 2001 From: Kyle Lucke Date: Tue, 19 Nov 2024 11:40:39 -0800 Subject: [PATCH] Remove unneeded use of gpu_types.h in topk_kernel_test.cc. PiperOrigin-RevId: 698091404 --- xla/service/gpu/kernels/BUILD | 1 - xla/service/gpu/kernels/topk_kernel_test.cc | 2 -- 2 files changed, 3 deletions(-) diff --git a/xla/service/gpu/kernels/BUILD b/xla/service/gpu/kernels/BUILD index e80992c72dd6ec..431b73d9319e49 100644 --- a/xla/service/gpu/kernels/BUILD +++ b/xla/service/gpu/kernels/BUILD @@ -197,7 +197,6 @@ xla_test( "//xla/stream_executor:platform_manager", "//xla/stream_executor:stream", "//xla/stream_executor/gpu:gpu_init", - "//xla/stream_executor/gpu:gpu_types_header", "//xla/stream_executor/host:host_platform", "@com_google_absl//absl/log:check", "@com_google_absl//absl/random", diff --git a/xla/service/gpu/kernels/topk_kernel_test.cc b/xla/service/gpu/kernels/topk_kernel_test.cc index 48bcaf07d06cd0..8ce9d80af12615 100644 --- a/xla/service/gpu/kernels/topk_kernel_test.cc +++ b/xla/service/gpu/kernels/topk_kernel_test.cc @@ -29,7 +29,6 @@ limitations under the License. #include "absl/time/time.h" #include "xla/stream_executor/device_memory_handle.h" #include "xla/stream_executor/gpu/gpu_init.h" -#include "xla/stream_executor/gpu/gpu_types.h" #include "xla/stream_executor/platform.h" #include "xla/stream_executor/platform_manager.h" #include "xla/stream_executor/stream.h" @@ -42,7 +41,6 @@ limitations under the License. namespace xla::gpu { namespace { -using se::gpu::GpuStreamHandle; using ::testing::Combine; using ::testing::Values;