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