use __HIPCC__ for device code

2024-01-24 11:30:01 -08:00
parent 1e0ccf7b87
commit 069cf1d019
23 changed files with 68 additions and 72 deletions
--- a/src/tree/split_evaluator.h
+++ b/src/tree/split_evaluator.h
@@ -124,7 +124,7 @@ class TreeEvaluator {
    [[nodiscard]] XGBOOST_DEVICE float Divide(float a, float b) const {
 #ifdef __CUDA_ARCH__
      return __fdividef(a, b);
-#elif defined(__HIP_PLATFORM_AMD__)
+#elif defined(__HIPCC__)
      return a / b;
 #else
      return a / b;