Remove unneeded use of gpu_types.h in topk_kernel_test.cc.

PiperOrigin-RevId: 698091404
openxla · Nov 19, 2024 · 4dae650 · 4dae650
1 parent e62fe73
commit 4dae650
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git 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
@@ -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;