megatron.dist_signal_handler.all_gather_item

megatron.dist_signal_handler.all_gather_item(item, dtype, group=None, async_op=False, local_rank=None)