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

add HIP flags, c_api

Browse Source
This commit is contained in:
amdsc21 2023-03-08 01:34:37 +01:00
parent f5f800c80d
commit 1e1c7fd8d5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/c_api/c_api_utils.h
View File

@ -173,7 +173,7 @@ inline float GetMissing(Json const &config) {
// Safe guard some global variables from being changed by XGBoost. // Safe guard some global variables from being changed by XGBoost.
class XGBoostAPIGuard { class XGBoostAPIGuard {
#if defined(XGBOOST_USE_CUDA) #if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
int32_t device_id_ {0}; int32_t device_id_ {0};
void SetGPUAttribute(); void SetGPUAttribute();