..
kit
[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert ( #5088 )
2023-11-28 16:54:42 +08:00
test_analyzer
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
test_auto_parallel
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
test_autochunk
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
test_booster
[plugin]fix 3d checkpoint load when booster boost without optimizer. ( #5135 )
2023-11-30 18:37:47 +08:00
test_checkpoint_io
[gemini] gemini support extra-dp ( #5043 )
2023-11-16 21:03:04 +08:00
test_cluster
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
test_config
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
test_device
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
test_fx
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
test_gptq
[feature] add gptq for inference ( #4754 )
2023-09-22 11:02:50 +08:00
test_infer
[Hotfix] Fix model policy matching strategy in ShardFormer ( #5064 )
2023-11-22 11:19:39 +08:00
test_infer_ops /triton
[inference] Refactor inference architecture ( #5057 )
2023-11-19 21:05:05 +08:00
test_lazy
[lazy] support from_pretrained ( #4801 )
2023-09-26 11:04:11 +08:00
test_legacy
[npu] add npu support for gemini and zero ( #5067 )
2023-11-20 16:12:41 +08:00
test_moe
[hotfix]: modify create_ep_hierarchical_group and add test ( #5032 )
2023-11-17 10:53:00 +08:00
test_optimizer
[test] merge old components to test to model zoo ( #4945 )
2023-10-20 10:35:08 +08:00
test_pipeline
[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert ( #5088 )
2023-11-28 16:54:42 +08:00
test_shardformer
[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert ( #5088 )
2023-11-28 16:54:42 +08:00
test_smoothquant
[inference] Add smmoothquant for llama ( #4904 )
2023-10-16 11:28:44 +08:00
test_tensor
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
test_utils
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
test_zero
[npu] add npu support for gemini and zero ( #5067 )
2023-11-20 16:12:41 +08:00
__init__.py
[zero] Update sharded model v2 using sharded param v2 ( #323 )
2022-03-11 15:50:28 +08:00
2a2ec49aa7