| .. |
|
kit
|
[example]add gpt2 benchmark example script. (#5295)
|
2024-03-04 16:18:13 +08:00 |
|
test_analyzer
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
|
test_auto_parallel
|
[npu] change device to accelerator api (#5239)
|
2024-01-09 10:20:05 +08:00 |
|
test_autochunk
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
|
test_booster
|
[shardformer] fix gathering output when using tensor parallelism (#5431)
|
2024-03-18 15:55:11 +08:00 |
|
test_checkpoint_io
|
[devops] fix compatibility (#5444)
|
2024-03-13 15:24:13 +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_lazy
|
[example]add gpt2 benchmark example script. (#5295)
|
2024-03-04 16:18:13 +08:00 |
|
test_legacy
|
[npu] change device to accelerator api (#5239)
|
2024-01-09 10:20:05 +08:00 |
|
test_moe
|
[moe] fix tests
|
2024-02-08 12:46:37 +08:00 |
|
test_optimizer
|
[lr-scheduler] fix load state dict and add test (#5369)
|
2024-02-06 14:23:32 +08:00 |
|
test_pipeline
|
Merge branch 'main' into sync/npu
|
2024-01-18 12:05:21 +08:00 |
|
test_shardformer
|
[devops] fix compatibility (#5444)
|
2024-03-13 15:24:13 +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
|
[feat] refactored extension module (#5298)
|
2024-01-25 17:01:48 +08:00 |
|
test_zero
|
[npu] change device to accelerator api (#5239)
|
2024-01-09 10:20:05 +08:00 |
|
__init__.py
|
…
|
|
5e16bf7980