From e74b3bbf3cf2120c1f8a9703dc241b9aeea542ff Mon Sep 17 00:00:00 2001
From: amdsc21 <96135754+amdsc21@users.noreply.github.com>
Date: Sat, 25 Mar 2023 05:17:39 +0100
Subject: [PATCH] fix macro

---
 src/metric/rank_metric.cu | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/metric/rank_metric.cu b/src/metric/rank_metric.cu
index 113857439..58ef9184e 100644
--- a/src/metric/rank_metric.cu
+++ b/src/metric/rank_metric.cu
@@ -24,6 +24,12 @@
 #include "xgboost/logging.h"             // for CHECK
 #include "xgboost/metric.h"
 
+#if defined(XGBOOST_USE_HIP)
+#include <hipcub/hipcub.hpp>
+
+namespace cub = hipcub;
+#endif
+
 namespace xgboost::metric {
 // tag the this file, used by force static link later.
 DMLC_REGISTRY_FILE_TAG(rank_metric_gpu);