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

2023-06-13 12:53:03 -07:00
parent c2f0486d37
commit e70810be8a
11 changed files with 190 additions and 106 deletions
--- a/tests/cpp/collective/test_nccl_device_communicator.cu
+++ b/tests/cpp/collective/test_nccl_device_communicator.cu
@@ -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 {