mirror of https://github.com/InternLM/InternLM
33 lines
884 B
Python
33 lines
884 B
Python
from .p2p import (
|
|
AsynCommunicator,
|
|
recv_backward,
|
|
recv_forward,
|
|
send_backward,
|
|
send_backward_and_recv_next_backward_async,
|
|
send_backward_recv_backward,
|
|
send_backward_recv_forward,
|
|
send_forward,
|
|
send_forward_and_recv_next_forward_async,
|
|
send_forward_backward_recv_forward_backward,
|
|
send_forward_recv_backward,
|
|
send_forward_recv_forward,
|
|
)
|
|
from .utils import recv_obj_meta, send_obj_meta
|
|
|
|
__all__ = [
|
|
"send_forward",
|
|
"send_forward_recv_forward",
|
|
"send_forward_backward_recv_forward_backward",
|
|
"send_backward",
|
|
"send_backward_recv_backward",
|
|
"send_backward_recv_forward",
|
|
"send_forward_recv_backward",
|
|
"recv_backward",
|
|
"recv_forward",
|
|
"send_obj_meta",
|
|
"recv_obj_meta",
|
|
"send_backward_and_recv_next_backward_async",
|
|
"send_forward_and_recv_next_forward_async",
|
|
"AsynCommunicator",
|
|
]
|