Skip to content

Commit b8644d6

Browse files
kluckeGoogle-ML-Automation
authored andcommitted
Remove unneeded use of gpu_types.h in topk_kernel_test.cc.
PiperOrigin-RevId: 698174417
1 parent d38eb56 commit b8644d6

File tree

2 files changed

+0
-3
lines changed

2 files changed

+0
-3
lines changed

xla/service/gpu/kernels/BUILD

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,6 @@ xla_test(
197197
"//xla/stream_executor:platform_manager",
198198
"//xla/stream_executor:stream",
199199
"//xla/stream_executor/gpu:gpu_init",
200-
"//xla/stream_executor/gpu:gpu_types_header",
201200
"//xla/stream_executor/host:host_platform",
202201
"@com_google_absl//absl/log:check",
203202
"@com_google_absl//absl/random",

xla/service/gpu/kernels/topk_kernel_test.cc

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@ limitations under the License.
2929
#include "absl/time/time.h"
3030
#include "xla/stream_executor/device_memory_handle.h"
3131
#include "xla/stream_executor/gpu/gpu_init.h"
32-
#include "xla/stream_executor/gpu/gpu_types.h"
3332
#include "xla/stream_executor/platform.h"
3433
#include "xla/stream_executor/platform_manager.h"
3534
#include "xla/stream_executor/stream.h"
@@ -42,7 +41,6 @@ limitations under the License.
4241
namespace xla::gpu {
4342
namespace {
4443

45-
using se::gpu::GpuStreamHandle;
4644
using ::testing::Combine;
4745
using ::testing::Values;
4846

0 commit comments

Comments
 (0)