From 58a9fe07b642ab178c016765e2c2aa4e6c40c6e0 Mon Sep 17 00:00:00 2001
From: amdsc21 <96135754+amdsc21@users.noreply.github.com>
Date: Fri, 10 Mar 2023 06:35:06 +0100
Subject: [PATCH] finish multiclass_obj.cu

---
 src/objective/multiclass_obj.cc  | 2 +-
 src/objective/multiclass_obj.cu  | 4 ++--
 src/objective/multiclass_obj.hip | 4 ++++
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/objective/multiclass_obj.cc b/src/objective/multiclass_obj.cc
index ec6616034..cfe088e9c 100644
--- a/src/objective/multiclass_obj.cc
+++ b/src/objective/multiclass_obj.cc
@@ -13,6 +13,6 @@ DMLC_REGISTRY_FILE_TAG(multiclass_obj);
 }  // namespace obj
 }  // namespace xgboost
 
-#ifndef XGBOOST_USE_CUDA
+#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
 #include "multiclass_obj.cu"
 #endif  // XGBOOST_USE_CUDA
diff --git a/src/objective/multiclass_obj.cu b/src/objective/multiclass_obj.cu
index 312992ec5..129685a19 100644
--- a/src/objective/multiclass_obj.cu
+++ b/src/objective/multiclass_obj.cu
@@ -24,9 +24,9 @@
 namespace xgboost {
 namespace obj {
 
-#if defined(XGBOOST_USE_CUDA)
+#if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
 DMLC_REGISTRY_FILE_TAG(multiclass_obj_gpu);
-#endif  // defined(XGBOOST_USE_CUDA)
+#endif  // defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
 
 struct SoftmaxMultiClassParam : public XGBoostParameter<SoftmaxMultiClassParam> {
   int num_class;
diff --git a/src/objective/multiclass_obj.hip b/src/objective/multiclass_obj.hip
index e69de29bb..82c7a2c06 100644
--- a/src/objective/multiclass_obj.hip
+++ b/src/objective/multiclass_obj.hip
@@ -0,0 +1,4 @@
+
+#if defined(XGBOOST_USE_HIP)
+#incldue "multiclass_obj.cu"
+#endif