add HIP flags

2023-03-08 01:22:25 +01:00
parent 75712b9c3c
commit 6b7be96373
2 changed files with 5 additions and 5 deletions
--- a/src/collective/communicator.h
+++ b/src/collective/communicator.h
@@ -98,7 +98,7 @@ class Communicator {
  /** @brief Get the communicator instance. */
  static Communicator *Get() { return communicator_.get(); }

-#if defined(XGBOOST_USE_CUDA)
+#if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
  /**
   * @brief Get the device communicator.
   *