add HIP flags

2023-03-08 01:33:38 +01:00
parent 6b7be96373
commit f5f800c80d
7 changed files with 20 additions and 20 deletions
--- a/src/tree/split_evaluator.h
+++ b/src/tree/split_evaluator.h
@@ -121,7 +121,7 @@ class TreeEvaluator {

    // Fast floating point division instruction on device
    XGBOOST_DEVICE float Divide(float a, float b) const {
-#ifdef __CUDA_ARCH__
+#if defined(__CUDA_ARCH__) || defined(__HIP_PLATFORM_AMD__)
      return __fdividef(a, b);
 #else
      return a / b;