megatron.text_generation.communication.recv_from_prev_pipeline_rank_#

megatron.text_generation.communication.recv_from_prev_pipeline_rank_(recv_buffer=None)#

Receive from previous pipeline stage and update the input buffer inplace.