From 1530c03f7d76434c19f23d74bf3ab16940f1f724 Mon Sep 17 00:00:00 2001
From: amdsc21 <96135754+amdsc21@users.noreply.github.com>
Date: Thu, 9 Mar 2023 22:43:51 +0100
Subject: [PATCH] finish constraints.cu

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

diff --git a/src/tree/constraints.cu b/src/tree/constraints.cu
index b6db0eda0..1065b9689 100644
--- a/src/tree/constraints.cu
+++ b/src/tree/constraints.cu
@@ -14,7 +14,12 @@
 #include "xgboost/span.h"
 #include "constraints.cuh"
 #include "param.h"
+
+#if defined(XGBOOST_USE_hip.CUDA)
 #include "../common/device_helpers.cuh"
+#elif defined(XGBOOST_USE_HIP)
+#include "../common/device_helpers.hip.h"
+#endif
 
 namespace xgboost {
 
diff --git a/src/tree/constraints.hip b/src/tree/constraints.hip
index e69de29bb..b8d6208cf 100644
--- a/src/tree/constraints.hip
+++ b/src/tree/constraints.hip
@@ -0,0 +1,4 @@
+
+#if defined(XGBOOST_USE_HIP)
+#include "constraints.cu"
+#endif