Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

enable ROCm support, rm un-necessary code

Browse Source
This commit is contained in:
Hui Liu 2023-10-30 12:39:30 -07:00
parent d0774a78e4
commit 1ec57fd1a3
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/common/transform.h
View File

@ -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));