mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-05-29 10:24:46 +00:00
b167258b6a
* [CLI] add CLI launcher
* Revert "[CLI] add CLI launcher"
This reverts commit df7e6506d4.
* refactor ppschedule to support tensor list
* polish
27 lines
865 B
Python
27 lines
865 B
Python
from .collective import all_gather, reduce_scatter, all_reduce, broadcast, reduce
|
|
from .p2p import (send_forward, send_forward_recv_forward, send_backward_recv_forward, send_backward,
|
|
send_backward_recv_backward, send_forward_recv_backward, send_forward_backward_recv_forward_backward,
|
|
recv_forward, recv_backward)
|
|
from .ring import ring_forward
|
|
from .utils import send_obj_meta, recv_obj_meta
|
|
|
|
__all__ = [
|
|
'all_gather',
|
|
'reduce_scatter',
|
|
'all_reduce',
|
|
'broadcast',
|
|
'reduce',
|
|
'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',
|
|
'ring_forward',
|
|
'send_obj_meta',
|
|
'recv_obj_meta',
|
|
]
|