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

rm device_helpers.hip.h from cu

Browse Source
This commit is contained in:
amdsc21
2023-03-26 00:24:11 +01:00
parent ee582f03c3
commit 7ee4734d3a
27 changed files with 14 additions and 167 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/metric/survival_metric.cu
View File

@@ -19,15 +19,10 @@
#include "xgboost/json.h"
#include "xgboost/metric.h"
#if defined(XGBOOST_USE_CUDA)
#if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
#include <thrust/execution_policy.h> // thrust::cuda::par
#include "../common/device_helpers.cuh"
#endif // XGBOOST_USE_CUDA
#if defined(XGBOOST_USE_HIP)
#include <thrust/execution_policy.h> // thrust::hip::par
#include "../common/device_helpers.hip.h"
#endif // XGBOOST_USE_HIP
#endif // XGBOOST_USE_CUDA || XGBOOST_USE_HIP
using AFTParam = xgboost::common::AFTParam;
using ProbabilityDistributionType = xgboost::common::ProbabilityDistributionType;