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

fix helpers.h, enable HIP

Browse Source
This commit is contained in:
amdsc21 2023-05-02 20:04:23 +02:00
parent 83e6fceb5c
commit 4a24ca2f95
2 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/cpp/helpers.h
View File

@ -39,13 +39,13 @@
#define GPUIDX -1 #define GPUIDX -1
#endif #endif
#if defined(__CUDACC__) #if defined(__CUDACC__) || defined(__HIP_PLATFORM_AMD__)
#define DeclareUnifiedDistributedTest(name) MGPU ## name #define DeclareUnifiedDistributedTest(name) MGPU ## name
#else #else
#define DeclareUnifiedDistributedTest(name) name #define DeclareUnifiedDistributedTest(name) name
#endif #endif
#if defined(__CUDACC__) #if defined(__CUDACC__) || defined(__HIP_PLATFORM_AMD__)
#define WORLD_SIZE_FOR_TEST (xgboost::common::AllVisibleGPUs()) #define WORLD_SIZE_FOR_TEST (xgboost::common::AllVisibleGPUs())
#else #else
#define WORLD_SIZE_FOR_TEST (3) #define WORLD_SIZE_FOR_TEST (3)

4
tests/cpp/objective/test_ranking_obj_gpu.hip
View File

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