| .. |
|
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
|
add scatter/gather optim for pipeline (#123)
|
2022-01-07 13:22:22 +08:00 |
|
context
|
update default logger (#100) (#101)
|
2022-01-04 20:03:26 +08:00 |
|
engine
|
add scatter/gather optim for pipeline (#123)
|
2022-01-07 13:22:22 +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 a bug in timer (#114)
|
2022-01-05 16:07:06 +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
|
add scatter/gather optim for pipeline (#123)
|
2022-01-07 13:22:22 +08:00 |
293fb40c42