Frank Lee 7cfed5f076 [feat] refactored extension module (#5298) ... * [feat] refactored extension module * polish * polish * polish * polish * polish * polish * polish * polish * polish * polish		2024-01-25 17:01:48 +08:00
..
model	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
multi_tensor_apply	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
rank_recorder	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
tensor_detector	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
__init__.py	[feat] refactored extension module (#5298)	2024-01-25 17:01:48 +08:00
common.py	[feat] refactored extension module (#5298)	2024-01-25 17:01:48 +08:00
memory.py	[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert (#5088)	2023-11-28 16:54:42 +08:00
timer.py	[npu] change device to accelerator api (#5239)	2024-01-09 10:20:05 +08:00