From 1e09c21456719f6b6cda76869929c0cd5605e24f Mon Sep 17 00:00:00 2001
From: amdsc21 <96135754+amdsc21@users.noreply.github.com>
Date: Thu, 9 Mar 2023 21:31:00 +0100
Subject: [PATCH] finished feature_groups.cu

---
 src/tree/gpu_hist/feature_groups.cu  | 5 +++++
 src/tree/gpu_hist/feature_groups.hip | 4 ++++
 2 files changed, 9 insertions(+)

diff --git a/src/tree/gpu_hist/feature_groups.cu b/src/tree/gpu_hist/feature_groups.cu
index 27ed9bd91..696c50bdb 100644
--- a/src/tree/gpu_hist/feature_groups.cu
+++ b/src/tree/gpu_hist/feature_groups.cu
@@ -8,7 +8,12 @@
 
 #include "feature_groups.cuh"
 
+#if defined(XGBOOST_USE_CUDA)
 #include "../../common/device_helpers.cuh"
+#elif defined(XGBOOST_USE_HIP)
+#include "../../common/device_helpers.hip.h"
+#endif
+
 #include "../../common/hist_util.h"
 
 namespace xgboost {
diff --git a/src/tree/gpu_hist/feature_groups.hip b/src/tree/gpu_hist/feature_groups.hip
index e69de29bb..ebc9aa533 100644
--- a/src/tree/gpu_hist/feature_groups.hip
+++ b/src/tree/gpu_hist/feature_groups.hip
@@ -0,0 +1,4 @@
+
+#if defined(XGBOOST_USE_HIP)
+#include "feature_groups.cu"
+#endif