add hip.h

2023-05-20 01:25:33 +02:00
parent 7663d47383
commit b22644fc10
44 changed files with 249 additions and 0 deletions
--- a/tests/cpp/objective/test_lambdarank_obj.cu
+++ b/tests/cpp/objective/test_lambdarank_obj.cu
@@ -7,8 +7,13 @@
 #include <cstdint>                               // for uint32_t
 #include <vector>                                // for vector

+#if defined(XGBOOST_USE_CUDA)
 #include "../../../src/common/cuda_context.cuh"  // for CUDAContext
 #include "../../../src/objective/lambdarank_obj.cuh"
+#elif defined(XGBOOST_USE_HIP)
+#include "../../../src/common/cuda_context.hip.h"  // for CUDAContext
+#include "../../../src/objective/lambdarank_obj.hip.h"
+#endif
 #include "test_lambdarank_obj.h"

 namespace xgboost::obj {