Hongxin Liu 58d8b8a2dd [misc] fit torch api upgradation and remove legecy import (#6093) ... * [amp] fit torch's new api * [amp] fix api call * [amp] fix api call * [misc] fit torch pytree api upgrade * [misc] remove legacy import * [misc] fit torch amp api * [misc] fit torch amp api		2024-10-18 16:48:52 +08:00
..
schedule	[misc] fit torch api upgradation and remove legecy import (#6093)	2024-10-18 16:48:52 +08:00
__init__.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
p2p.py	[FP8] rebase main (#5963)	2024-08-06 16:29:37 +08:00
stage_manager.py	[Feature] optimize PP overlap (#5735)	2024-06-26 14:48:02 +08:00