8 Commits (bce9499ed33f7a8359bbb568c7ee18d72e8aa731)

Author	SHA1	Message	Date
Xuanlei Zhao	3acbf6d496	[npu] add npu support for hybrid plugin and llama (#5090) ... * llama 3d * update * fix autocast	1 year ago
Hongxin Liu	079bf3cb26	[misc] update pre-commit and run all files (#4752) ... * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format	1 year ago
Jianghai	b366f1d99f	[NFC] Fix format for mixed precision (#4253) ... * [NFC] polish colossalai/booster/mixed_precision/mixed_precision_base.py code style	1 year ago
Wenhao Chen	725af3eeeb	[booster] make optimizer argument optional for boost (#3993) ... * feat: make optimizer optional in Booster.boost * test: skip unet test if diffusers version > 0.10.2	1 year ago
jiangmingyan	e871e342b3	[API] add docstrings and initialization to apex amp, naive amp (#3783) ... * [mixed_precison] add naive amp demo * [mixed_precison] add naive amp demo * [api] add docstrings and initialization to apex amp, naive amp * [api] add docstring to apex amp/ naive amp * [api] add docstring to apex amp/ naive amp * [api] add docstring to apex amp/ naive amp * [api] add docstring to apex amp/ naive amp * [api] add docstring to apex amp/ naive amp * [api] add docstring to apex amp/ naive amp * [api] fix * [api] fix	2 years ago
jiangmingyan	2703a37ac9	[amp] Add naive amp demo (#3774) ... * [mixed_precison] add naive amp demo * [mixed_precison] add naive amp demo	2 years ago
Frank Lee	73d3e4d309	[booster] implemented the torch ddd + resnet example (#3232) ... * [booster] implemented the torch ddd + resnet example * polish code	2 years ago
Frank Lee	ed19290560	[booster] implemented mixed precision class (#3151) ... * [booster] implemented mixed precision class * polish code	2 years ago