github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/pipeline
History
Kirigaya Kazuto 318fbf1145
[NFC] polish colossalai/utils/multi_tensor_apply/multi_tensor_apply.py code style (#1559) 		2022-09-08 22:04:34 +08:00
..
rpc [NFC] polish colossalai/utils/multi_tensor_apply/multi_tensor_apply.py code style (#1559) 2022-09-08 22:04:34 +08:00
__init__.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
layer_sepc.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
pipelinable.py [pipeline]add customized policy (#1139) 2022-06-21 15:23:41 +08:00
pipeline_process_group.py [pipeline/pipleline_process_group] finish PipelineProcessGroup to manage local abd global rank in TP,DP and PP (#1508) 2022-09-01 17:45:47 +08:00
utils.py [pipeline]add customized policy (#1139) 2022-06-21 15:23:41 +08:00