restore device helper

tests/cpp/common/test_bitfield.cu

@@ -6,7 +6,11 @@
 #include <thrust/device_vector.h>
 #include <vector>
 #include "../../../src/common/bitfield.h"
+#if defined(XGBOOST_USE_CUDA)
 #include "../../../src/common/device_helpers.cuh"
+#elif defined(XGBOOST_USE_HIP)
+#include "../../../src/common/device_helpers.hip.h"
+#endif
 
 namespace xgboost {
 

tests/cpp/common/test_device_helpers.cu

@@ -6,7 +6,11 @@
 #include <thrust/device_vector.h>
 #include <vector>
 #include <xgboost/base.h>
+#if defined(XGBOOST_USE_CUDA)
 #include "../../../src/common/device_helpers.cuh"
+#elif defined(XGBOOST_USE_HIP)
+#include "../../../src/common/device_helpers.hip.h"
+#endif
 #include "../../../src/common/quantile.h"
 #include "../helpers.h"
 #include "gtest/gtest.h"

tests/cpp/common/test_gpu_compressed_iterator.cu

@@ -1,5 +1,9 @@
 #include "../../../src/common/compressed_iterator.h"
+#if defined(XGBOOST_USE_CUDA)
 #include "../../../src/common/device_helpers.cuh"
+#elif defined(XGBOOST_USE_HIP)
+#include "../../../src/common/device_helpers.hip.h"
+#endif
 #include "gtest/gtest.h"
 #include <algorithm>
 #include <thrust/device_vector.h>

tests/cpp/common/test_host_device_vector.cu

@@ -6,7 +6,11 @@
 #include <thrust/equal.h>
 #include <thrust/iterator/counting_iterator.h>
 
+#if defined(XGBOOST_USE_CUDA)
 #include "../../../src/common/device_helpers.cuh"
+#elif defined(XGBOOST_USE_HIP)
+#include "../../../src/common/device_helpers.hip.h"
+#endif
 #include <xgboost/host_device_vector.h>
 
 namespace xgboost {

tests/cpp/common/test_span.cu

@@ -7,7 +7,11 @@
 #include <thrust/device_vector.h>
 #include <thrust/execution_policy.h>
 
+#if defined(XGBOOST_USE_CUDA)
 #include "../../../src/common/device_helpers.cuh"
+#elif defined(XGBOOST_USE_HIP)
+#include "../../../src/common/device_helpers.hip.h"
+#endif
 
 #include <xgboost/span.h>
 #include "test_span.h"