From 1ec57fd1a36779e125df73930f1fa11784e8611b Mon Sep 17 00:00:00 2001
From: Hui Liu <96135754+amdsc21@users.noreply.github.com>
Date: Mon, 30 Oct 2023 12:39:30 -0700
Subject: [PATCH] enable ROCm support, rm un-necessary code

---
 src/common/transform.h | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/src/common/transform.h b/src/common/transform.h
index fd6f82817..1a200323f 100644
--- a/src/common/transform.h
+++ b/src/common/transform.h
@@ -143,11 +143,7 @@ class Transform {
       size_t shard_size = range_size;
       Range shard_range {0, static_cast<Range::DifferenceType>(shard_size)};
 
-#if defined(XGBOOST_USE_HIP)
-      dh::safe_cuda(hipSetDevice(device_));
-#elif defined(XGBOOST_USE_CUDA)
       dh::safe_cuda(cudaSetDevice(device_));
-#endif
 
       const int kGrids =
           static_cast<int>(DivRoundUp(*(range_.end()), kBlockThreads));