Adds support for allgather_into_tensor_coalesced
and reduce_scatter_tensor_coalesced
#351
Loading
allgather_into_tensor_coalesced
and reduce_scatter_tensor_coalesced
#351