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

finish ellpack_page_source.cu

Browse Source
This commit is contained in:
amdsc21 2023-03-10 05:06:36 +01:00
parent fa9f69dd85
commit 61c0b19331
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/data/ellpack_page_source.cu
View File

@ -10,7 +10,11 @@
namespace xgboost { namespace xgboost {
namespace data { namespace data {
void EllpackPageSource::Fetch() { void EllpackPageSource::Fetch() {
#if defined(XGBOOST_USE_CUDA)
dh::safe_cuda(cudaSetDevice(param_.gpu_id)); dh::safe_cuda(cudaSetDevice(param_.gpu_id));
#elif defined(XGBOOST_USE_HIP)
dh::safe_cuda(hipSetDevice(param_.gpu_id));
#endif
if (!this->ReadCache()) { if (!this->ReadCache()) {
if (count_ != 0 && !sync_) { if (count_ != 0 && !sync_) {
// source is initialized to be the 0th page during construction, so when count_ is 0 // source is initialized to be the 0th page during construction, so when count_ is 0

4
src/data/ellpack_page_source.hip
View File

@ -0,0 +1,4 @@
#if defined(XGBOOST_USE_HIP)
#include "ellpack_page_source.cu"
#endif