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

add context.hip

Browse Source
This commit is contained in:
amdsc21 2023-03-08 07:34:19 +01:00
parent 7a3a9b682a
commit bdcb036592
4 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/common/cuda_context.cuh
View File

@ -5,7 +5,11 @@
#define XGBOOST_COMMON_CUDA_CONTEXT_CUH_ #define XGBOOST_COMMON_CUDA_CONTEXT_CUH_
#include <thrust/execution_policy.h> #include <thrust/execution_policy.h>
#if defined(XGBOOST_USE_HIP)
#include "device_helpers.hip.h"
#elif defined(XGBOOST_USE_CUDA)
#include "device_helpers.cuh" #include "device_helpers.cuh"
#endif
namespace xgboost { namespace xgboost {
struct CUDAContext { struct CUDAContext {

2
src/common/device_helpers.hip.h
View File

@ -23,7 +23,7 @@
#include <chrono> #include <chrono>
#include <cstddef> // for size_t #include <cstddef> // for size_t
#include <hipcub/hipcub.hpp> #include <hipcub/hipcub.hpp>
#include <cub/util_allocator.cuh> #include <hipcub/util_allocator.hpp>
#include <numeric> #include <numeric>
#include <sstream> #include <sstream>
#include <string> #include <string>

8
src/context.cc
View File

@ -18,7 +18,7 @@ std::int64_t constexpr Context::kDefaultSeed;
Context::Context() : cfs_cpu_count_{common::GetCfsCPUCount()} {} Context::Context() : cfs_cpu_count_{common::GetCfsCPUCount()} {}
void Context::ConfigureGpuId(bool require_gpu) { void Context::ConfigureGpuId(bool require_gpu) {
#if defined(XGBOOST_USE_CUDA) #if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
if (gpu_id == kCpuId) { // 0. User didn't specify the `gpu_id' if (gpu_id == kCpuId) { // 0. User didn't specify the `gpu_id'
if (require_gpu) { // 1. `tree_method' or `predictor' or both are using if (require_gpu) { // 1. `tree_method' or `predictor' or both are using
// GPU. // GPU.
@ -47,7 +47,7 @@ void Context::ConfigureGpuId(bool require_gpu) {
// Just set it to CPU, don't think about it. // Just set it to CPU, don't think about it.
this->UpdateAllowUnknown(Args{{"gpu_id", std::to_string(kCpuId)}}); this->UpdateAllowUnknown(Args{{"gpu_id", std::to_string(kCpuId)}});
(void)(require_gpu); (void)(require_gpu);
#endif // defined(XGBOOST_USE_CUDA) #endif // defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_
common::SetDevice(this->gpu_id); common::SetDevice(this->gpu_id);
} }
@ -60,10 +60,10 @@ std::int32_t Context::Threads() const {
return n_threads; return n_threads;
} }
#if !defined(XGBOOST_USE_CUDA) #if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
CUDAContext const* Context::CUDACtx() const { CUDAContext const* Context::CUDACtx() const {
common::AssertGPUSupport(); common::AssertGPUSupport();
return nullptr; return nullptr;
} }
#endif // defined(XGBOOST_USE_CUDA) #endif // defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
} // namespace xgboost } // namespace xgboost

2
src/context.hip
View File

@ -0,0 +1,2 @@
#include "context.cu"