try elementwise_metric.cu

src/metric/elementwise_metric.cc

@@ -3,8 +3,6 @@
  */
 // Dummy file to keep the CUDA conditional compile trick.
 
-#if !defined(XGBOOST_USE_CUDA)
+#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
 #include "elementwise_metric.cu"
-#elif !defined(XGBOOST_USE_HIP)
-#include "elementwise_metric.hip"
-#endif  // !defined(XGBOOST_USE_CUDA)
+#endif  // !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)

src/metric/elementwise_metric.hip

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

src/metric/multiclass_metric.cc

@@ -3,6 +3,6 @@
  */
 // Dummy file to keep the CUDA conditional compile trick.
 
-#if !defined(XGBOOST_USE_CUDA)
+#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
 #include "multiclass_metric.cu"
 #endif  // !defined(XGBOOST_USE_CUDA)

src/metric/multiclass_metric.cu

@@ -23,6 +23,15 @@
 #include "../common/device_helpers.cuh"
 #endif  // XGBOOST_USE_CUDA
 
+#if defined(XGBOOST_USE_HIP)
+#include <thrust/execution_policy.h>  // thrust::cuda::par
+#include <thrust/functional.h>        // thrust::plus<>
+#include <thrust/iterator/counting_iterator.h>
+#include <thrust/transform_reduce.h>
+
+#include "../common/device_helpers.hip.h"
+#endif  // XGBOOST_USE_HIP
+
 namespace xgboost {
 namespace metric {
 // tag the this file, used by force static link later.