megengine.functional.distributed.collective_comm

collective_comm(inp, mode, group, device)[source]

Helper function for applying collective communication functions.

previous

megengine.functional.distributed.reduce_scatter_sum

next

megengine.functional.distributed.remote_send