diff --git a/include/dlaf/sender/transform_mpi.h b/include/dlaf/sender/transform_mpi.h index f5c2ac728a..c4f3f0b4fa 100644 --- a/include/dlaf/sender/transform_mpi.h +++ b/include/dlaf/sender/transform_mpi.h @@ -23,13 +23,18 @@ namespace dlaf::comm::internal { -/// This helper "consumes" a CommunicatorPipelineExclusiveWrapper ensuring that after this call -/// the one passed as argument gets destroyed. All other types left as they are -/// by the second overload. +/// This helper "consumes" a CommunicatorPipelineExclusiveWrapper or a CommunicatorPipelineSharedWrapper +/// ensuring that after this call the one passed as argument gets destroyed. All other types left as they +/// are by the generic overload. inline void consumeCommunicatorWrapper(CommunicatorPipelineExclusiveWrapper& comm_wrapper) { [[maybe_unused]] auto comm_wrapper_local = std::move(comm_wrapper); } +/// \overload consumeCommunicatorWrapper +inline void consumeCommunicatorWrapper(CommunicatorPipelineSharedWrapper& comm_wrapper) { + [[maybe_unused]] auto comm_wrapper_local = std::move(comm_wrapper); +} + /// \overload consumeCommunicatorWrapper template void consumeCommunicatorWrapper(T&) {}