ver217 7904baf6e1 fix layers/schedule for hybrid parallelization (#111) (#112)

2022-01-04 20:52:31 +08:00

..

amp

Optimize pipeline schedule (#94)

2021-12-30 15:56:46 +08:00

builder

Optimize pipeline schedule (#94)

2021-12-30 15:56:46 +08:00

communication

Layer integration (#83)

2021-12-27 15:04:32 +08:00

context

update default logger (#100) (#101)

2022-01-04 20:03:26 +08:00

engine

fix layers/schedule for hybrid parallelization (#111) (#112)

2022-01-04 20:52:31 +08:00

kernel

add colossalai kernel module (#55)

2021-12-21 12:19:52 +08:00

logging

update default logger (#100) (#101)

2022-01-04 20:03:26 +08:00

nn

fix layers/schedule for hybrid parallelization (#111) (#112)

2022-01-04 20:52:31 +08:00

registry

Develop/experiments (#59)

2021-12-09 15:08:29 +08:00

trainer

fix layers/schedule for hybrid parallelization (#111) (#112)

2022-01-04 20:52:31 +08:00

utils

Optimize pipeline schedule (#94)

2021-12-30 15:56:46 +08:00

zero

fixed zero level 3 dtype bug (#76)

2021-12-20 17:00:53 +08:00

__init__.py

Develop/experiments (#59)

2021-12-09 15:08:29 +08:00

constants.py

Optimize pipeline schedule (#94)

2021-12-30 15:56:46 +08:00

core.py

Develop/experiments (#59)

2021-12-09 15:08:29 +08:00

initialize.py

Optimize pipeline schedule (#94)

2021-12-30 15:56:46 +08:00