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

fix warp header

Browse Source
This commit is contained in:
amdsc21 2023-03-25 22:37:37 +01:00
parent 5098735698
commit 3ee3bea683
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/predictor/gpu_predictor.cu
View File

@ -1,7 +1,6 @@
/** /**
* Copyright 2017-2023 by XGBoost Contributors * Copyright 2017-2023 by XGBoost Contributors
*/ */
#include <amd_warp_primitives.h>
#include <GPUTreeShap/gpu_treeshap.h> #include <GPUTreeShap/gpu_treeshap.h>
#include <thrust/copy.h> #include <thrust/copy.h>
#include <thrust/device_ptr.h> #include <thrust/device_ptr.h>
@ -19,6 +18,7 @@
#if defined(XGBOOST_USE_CUDA) #if defined(XGBOOST_USE_CUDA)
#include "../common/device_helpers.cuh" #include "../common/device_helpers.cuh"
#elif defined(XGBOOST_USE_HIP) #elif defined(XGBOOST_USE_HIP)
#include <amd_warp_primitives.h>
#include "../common/device_helpers.hip.h" #include "../common/device_helpers.hip.h"
#endif #endif