ColossalAI/colossalai/amp
Frank Lee 8518263b80
[test] fixed the triton version for testing (#2608)
2023-02-07 13:49:38 +08:00
..
apex_amp [test] fixed the triton version for testing (#2608) 2023-02-07 13:49:38 +08:00
naive_amp
torch_amp
__init__.py
amp_type.py