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
..
__init__.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
chunk_memstats_collector.py	[npu] change device to accelerator api (#5239)	2024-01-09 10:20:05 +08:00
memory_monitor.py	[npu] change device to accelerator api (#5239)	2024-01-09 10:20:05 +08:00
memory_stats.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
memstats_collector.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
param_runtime_order.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
runtime_mem_tracer.py	[misc] fit torch api upgradation and remove legecy import (#6093)	2024-10-18 16:48:52 +08:00
static_memstats_collector.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
utils.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00