mirror of https://github.com/hpcaitech/ColossalAI
40d376c566
* [setup] support pre-build and jit-build of cuda kernels * polish code * polish code * polish code * polish code * polish code * polish code |
||
|---|---|---|
| .. | ||
| apex_amp | ||
| naive_amp | ||
| torch_amp | ||
| __init__.py | ||
| amp_type.py | ||