Refactor device communicator to make allreduce more flexible (#9295)

2023-06-14 03:53:03 +08:00
parent c2f0486d37
commit e70810be8a
11 changed files with 190 additions and 106 deletions
@@ -8,6 +8,7 @@
 #include <string>  // for string

 #include "../../../src/collective/nccl_device_communicator.cuh"
+#include "../../../src/collective/communicator-inl.cuh"

 namespace xgboost {
 namespace collective {