kurisusnowdeng 0b8161fab8 updated tp layers

2022-11-02 12:19:38 +08:00

..

__init__.py

[MOE] add unitest for MOE experts layout, gradient handler and kernel (#469)

2022-03-21 13:35:04 +08:00

initializer_1d.py

[model checkpoint] add gloo groups for cpu tensor communication (#589)

2022-04-01 10:15:52 +08:00

initializer_2d.py

[model checkpoint] add gloo groups for cpu tensor communication (#589)

2022-04-01 10:15:52 +08:00

initializer_2p5d.py

[usability] improved error messages in the context module (#856)

2022-04-25 13:42:31 +08:00

initializer_3d.py

updated tp layers

2022-11-02 12:19:38 +08:00

initializer_data.py

[NFC] polish <colossalai/context/process_group_initializer/initializer_data.py> code stype (#626)

2022-04-06 11:40:59 +08:00

initializer_model.py

[model checkpoint] add gloo groups for cpu tensor communication (#589)

2022-04-01 10:15:52 +08:00

initializer_pipeline.py

[model checkpoint] add gloo groups for cpu tensor communication (#589)

2022-04-01 10:15:52 +08:00

initializer_sequence.py

[NFC] polish colossalai/context/process_group_initializer/initializer_sequence.py colossalai/context/process_group_initializer initializer_tensor.py code style (#639)

2022-04-06 11:40:59 +08:00

initializer_tensor.py

[NFC] polish colossalai/context/process_group_initializer/initializer_sequence.py colossalai/context/process_group_initializer initializer_tensor.py code style (#639)

2022-04-06 11:40:59 +08:00

process_group_initializer.py

[NFC] polish colossalai/context/process_group_initializer/process_group_initializer.py code stype (#617)

2022-04-06 11:40:59 +08:00