diff --git a/cpp/include/raft/comms/detail/mpi_comms.hpp b/cpp/include/raft/comms/detail/mpi_comms.hpp index 1825da0233..c0bbc0ec40 100644 --- a/cpp/include/raft/comms/detail/mpi_comms.hpp +++ b/cpp/include/raft/comms/detail/mpi_comms.hpp @@ -127,13 +127,16 @@ class mpi_comms : public comms_iface { // initializing NCCL ncclConfig_t nccl_config = NCCL_CONFIG_INITIALIZER; - nccl_config.splitShare = 1; + nccl_config.splitShare = 1; RAFT_NCCL_TRY(ncclCommInitRankConfig(&nccl_comm_, size_, id, rank_, &nccl_config)); initialize(); } - mpi_comms(MPI_Comm mpi_comm, bool owns_mpi_comm, ncclComm_t nccl_comm, rmm::cuda_stream_view stream) + mpi_comms(MPI_Comm mpi_comm, + bool owns_mpi_comm, + ncclComm_t nccl_comm, + rmm::cuda_stream_view stream) : owns_mpi_comm_(owns_mpi_comm), mpi_comm_(mpi_comm), nccl_comm_(nccl_comm), diff --git a/cpp/include/raft/comms/detail/std_comms.hpp b/cpp/include/raft/comms/detail/std_comms.hpp index a5214fcf8b..0e48b73711 100644 --- a/cpp/include/raft/comms/detail/std_comms.hpp +++ b/cpp/include/raft/comms/detail/std_comms.hpp @@ -144,7 +144,8 @@ class std_comms : public comms_iface { RAFT_NCCL_TRY(ncclCommSplit(nccl_comm_, color, key, &new_nccl_comm, nullptr)); int new_nccl_comm_size{}; RAFT_NCCL_TRY(ncclCommCount(new_nccl_comm, &new_nccl_comm_size)); - return std::unique_ptr(new std_comms(new_nccl_comm, new_nccl_comm_size, key, stream_, true)); + return std::unique_ptr( + new std_comms(new_nccl_comm, new_nccl_comm_size, key, stream_, true)); } void barrier() const