fix auc.cu

2023-03-09 20:29:38 +01:00
parent 6eba0a56ec
commit a56055225a
7 changed files with 205 additions and 20 deletions
--- a/src/collective/device_communicator.cuh
+++ b/src/collective/device_communicator.cuh
@@ -4,7 +4,11 @@
 #pragma once
 #include <vector>

+#if defined(XGBOOST_USE_HIP)
+#include "../common/device_helpers.hip.h"
+#elif defined(XGBOOST_USE_CUDA)
 #include "../common/device_helpers.cuh"
+#endif

 namespace xgboost {
 namespace collective {