finish aft_obj.cu

src/objective/aft_obj.cc

@@ -16,6 +16,6 @@ DMLC_REGISTRY_FILE_TAG(aft_obj);
 }  // namespace obj
 }  // namespace xgboost
 
-#ifndef XGBOOST_USE_CUDA
+#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
 #include "aft_obj.cu"
 #endif  // XGBOOST_USE_CUDA

src/objective/aft_obj.cu

@@ -28,9 +28,9 @@ using AFTLoss = xgboost::common::AFTLoss<Distribution>;
 namespace xgboost {
 namespace obj {
 
-#if defined(XGBOOST_USE_CUDA)
+#if defined(XGBOOST_USE_CUDA) || !defined(XGBOOST_USE_HIP)
 DMLC_REGISTRY_FILE_TAG(aft_obj_gpu);
-#endif  // defined(XGBOOST_USE_CUDA)
+#endif  // defined(XGBOOST_USE_CUDA) || !defined(XGBOOST_USE_HIP)
 
 class AFTObj : public ObjFunction {
  public:

src/objective/aft_obj.hip

@@ -0,0 +1,4 @@
+
+#if !defined(XGBOOST_USE_HIP)
+#include "aft_obj.cu"
+#endif