1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-02 13:45:36 +00:00
ColossalAI/tests
flybird11111 2a0558d8ec
[ci] fix test_hybrid_parallel_plugin_checkpoint_io.py ()
* fix ci

fix

* fix test

* revert: revert p2p

* feat: add enable_metadata_cache option

* revert: enable t5 tests

* fix

---------

Co-authored-by: Wenhao Chen <cwher@outlook.com>
2024-01-17 13:38:55 +08:00
..
kit [workflow] fixed oom tests () 2024-01-16 18:55:13 +08:00
test_analyzer [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
test_auto_parallel [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
test_autochunk [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
test_booster [workflow] fixed oom tests () 2024-01-16 18:55:13 +08:00
test_checkpoint_io [ci] fix test_hybrid_parallel_plugin_checkpoint_io.py () 2024-01-17 13:38:55 +08:00
test_cluster [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
test_config [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
test_device [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
test_fx [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
test_gptq [feature] add gptq for inference () 2023-09-22 11:02:50 +08:00
test_infer [Hotfix] Fix model policy matching strategy in ShardFormer () 2023-11-22 11:19:39 +08:00
test_lazy [workflow] fixed oom tests () 2024-01-16 18:55:13 +08:00
test_legacy [npu] add npu support for gemini and zero () 2023-11-20 16:12:41 +08:00
test_moe [hotfix]: modify create_ep_hierarchical_group and add test () 2023-11-17 10:53:00 +08:00
test_optimizer [test] merge old components to test to model zoo () 2023-10-20 10:35:08 +08:00
test_pipeline [hotfix]: add pp sanity check and fix mbs arg () 2024-01-15 15:57:40 +08:00
test_shardformer [ci] fix shardformer tests. () 2024-01-11 19:07:45 +08:00
test_smoothquant [inference] Add smmoothquant for llama () 2023-10-16 11:28:44 +08:00
test_tensor [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
test_utils [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
test_zero [npu] add npu support for gemini and zero () 2023-11-20 16:12:41 +08:00
__init__.py [zero] Update sharded model v2 using sharded param v2 () 2022-03-11 15:50:28 +08:00