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

finish common.cu

Browse Source
This commit is contained in:
amdsc21 2023-03-10 05:19:41 +01:00
parent 8fd2af1c8b
commit 91a5ef762e
3 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/common/common.cu
View File

@ -8,7 +8,11 @@ namespace common {
void SetDevice(std::int32_t device) { void SetDevice(std::int32_t device) {
if (device >= 0) { if (device >= 0) {
#if defined(XGBOOST_USE_CUDA)
dh::safe_cuda(cudaSetDevice(device)); dh::safe_cuda(cudaSetDevice(device));
#elif defined(XGBOOST_USE_HIP)
dh::safe_cuda(hipSetDevice(device));
#endif
} }
} }
@ -17,9 +21,17 @@ int AllVisibleGPUs() {
try { try {
// When compiled with CUDA but running on CPU only device, // When compiled with CUDA but running on CPU only device,
// cudaGetDeviceCount will fail. // cudaGetDeviceCount will fail.
#if defined(XGBOOST_USE_CUDA)
dh::safe_cuda(cudaGetDeviceCount(&n_visgpus)); dh::safe_cuda(cudaGetDeviceCount(&n_visgpus));
#elif defined(XGBOOST_USE_HIP)
dh::safe_cuda(hipGetDeviceCount(&n_visgpus));
#endif
} catch (const dmlc::Error &) { } catch (const dmlc::Error &) {
#if defined(XGBOOST_USE_CUDA)
cudaGetLastError(); // reset error. cudaGetLastError(); // reset error.
#elif defined(XGBOOST_USE_HIP)
hipGetLastError(); // reset error.
#endif
return 0; return 0;
} }
return n_visgpus; return n_visgpus;

2
src/common/common.h
View File

@ -156,7 +156,7 @@ int AllVisibleGPUs();
inline void AssertGPUSupport() { inline void AssertGPUSupport() {
#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP) #if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
LOG(FATAL) << "XGBoost version not compiled with GPU support."; LOG(FATAL) << "XGBoost version not compiled with GPU support.";
#endif // XGBOOST_USE_CUDA #endif // XGBOOST_USE_CUDA && XGBOOST_USE_HIP
} }
inline void AssertOneAPISupport() { inline void AssertOneAPISupport() {

4
src/common/common.hip
View File

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