megatron.p2p_communication.recv_forward#

megatron.p2p_communication.recv_forward(tensor_shape=None, dtype_=None, timers=None)#

Receive tensor from previous rank in pipeline (forward receive).