mirror of https://github.com/hpcaitech/ColossalAI
ed19290560
* [booster] implemented mixed precision class * polish code |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| bf16.py | ||
| fp8.py | ||
| fp16_apex.py | ||
| fp16_torch.py | ||
| mixed_precision_base.py | ||