megatron.text_generation.communication.send_to_next_pipeline_rank

megatron.text_generation.communication.send_to_next_pipeline_rank(tensor=None)

Send output to the next pipeline stage.