From 4a24ca2f95966af5a8998b595de21399c09686bc Mon Sep 17 00:00:00 2001
From: amdsc21 <96135754+amdsc21@users.noreply.github.com>
Date: Tue, 2 May 2023 20:04:23 +0200
Subject: [PATCH] fix helpers.h, enable HIP

---
 tests/cpp/helpers.h                          | 4 ++--
 tests/cpp/objective/test_ranking_obj_gpu.hip | 4 ----
 2 files changed, 2 insertions(+), 6 deletions(-)
 delete mode 100644 tests/cpp/objective/test_ranking_obj_gpu.hip

diff --git a/tests/cpp/helpers.h b/tests/cpp/helpers.h
index bcd27c568..9d7cd5590 100644
--- a/tests/cpp/helpers.h
+++ b/tests/cpp/helpers.h
@@ -39,13 +39,13 @@
 #define GPUIDX -1
 #endif
 
-#if defined(__CUDACC__)
+#if defined(__CUDACC__) || defined(__HIP_PLATFORM_AMD__)
 #define DeclareUnifiedDistributedTest(name) MGPU ## name
 #else
 #define DeclareUnifiedDistributedTest(name) name
 #endif
 
-#if defined(__CUDACC__)
+#if defined(__CUDACC__) || defined(__HIP_PLATFORM_AMD__)
 #define WORLD_SIZE_FOR_TEST (xgboost::common::AllVisibleGPUs())
 #else
 #define WORLD_SIZE_FOR_TEST (3)
diff --git a/tests/cpp/objective/test_ranking_obj_gpu.hip b/tests/cpp/objective/test_ranking_obj_gpu.hip
deleted file mode 100644
index a39a4d006..000000000
--- a/tests/cpp/objective/test_ranking_obj_gpu.hip
+++ /dev/null
@@ -1,4 +0,0 @@
-
-#if defined(XGBOOST_USE_HIP)
-#include "test_ranking_obj_gpu.cu"
-#endif