ColossalAI/colossalai/amp/apex_amp at ac178ca5c17ca751ff9df38be81da2b4a005fc0d - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-04 09:52:50 +00:00

Files

History

Frank Lee 8518263b80 [test] fixed the triton version for testing (#2608 )

2023-02-07 13:49:38 +08:00

..

__init__.py

[NFC] polish colossalai/amp/apex_amp/__init__.py code style (#1853 )

2022-11-09 14:49:42 +08:00

apex_amp.py

[test] fixed the triton version for testing (#2608 )

2023-02-07 13:49:38 +08:00