mirror of https://github.com/hpcaitech/ColossalAI
6ae8ed0407
* [lazyinit] fix shared module * [tests] add lazy init test utils * [tests] add torchvision for lazy init * [lazyinit] fix pre op fn * [lazyinit] handle legacy constructor * [tests] refactor lazy init test models * [tests] refactor lazy init test utils * [lazyinit] fix ops don't support meta * [tests] lazy init test timm models * [lazyinit] fix set data * [lazyinit] handle apex layers * [tests] lazy init test transformers models * [tests] lazy init test torchaudio models * [lazyinit] fix import path * [tests] lazy init test torchrec models * [tests] update torch version in CI * [tests] revert torch version in CI * [tests] skip lazy init test |
||
|---|---|---|
| .. | ||
| _C | ||
| _analyzer | ||
| amp | ||
| auto_parallel | ||
| autochunk | ||
| booster | ||
| builder | ||
| cli | ||
| communication | ||
| context | ||
| device | ||
| engine | ||
| fx | ||
| gemini | ||
| kernel | ||
| logging | ||
| nn | ||
| pipeline | ||
| registry | ||
| tensor | ||
| testing | ||
| trainer | ||
| utils | ||
| zero | ||
| __init__.py | ||
| constants.py | ||
| core.py | ||
| global_variables.py | ||
| initialize.py | ||