ColossalAI/tests
digger yu 9c2feb2f0b
fix some typo with colossalai/device colossalai/tensor/ etc. (#4171)
Co-authored-by: flybird11111 <1829166702@qq.com>
2023-09-12 17:41:52 +08:00
..
components_to_test [legacy] move communication and nn to legacy and refactor logger (#4671) 2023-09-11 16:24:28 +08:00
kit fix some typo with colossalai/device colossalai/tensor/ etc. (#4171) 2023-09-12 17:41:52 +08:00
test_amp
test_analyzer
test_auto_parallel
test_autochunk
test_booster fix some typo with colossalai/device colossalai/tensor/ etc. (#4171) 2023-09-12 17:41:52 +08:00
test_checkpoint_io Merge branch 'main' into feature/shardformer 2023-09-05 21:54:08 +08:00
test_cluster
test_config [devops] add large-scale distributed test marker (#4452) 2023-08-16 18:56:52 +08:00
test_context [devops] add large-scale distributed test marker (#4452) 2023-08-16 18:56:52 +08:00
test_data [devops] add large-scale distributed test marker (#4452) 2023-08-16 18:56:52 +08:00
test_device
test_fx [misc] resolve code factor issues (#4433) 2023-08-15 23:25:14 +08:00
test_infer [Feature] The first PR to Add TP inference engine, kv-cache manager and related kernels for our inference system (#4577) 2023-09-12 01:22:56 +08:00
test_infer_ops [Feature] The first PR to Add TP inference engine, kv-cache manager and related kernels for our inference system (#4577) 2023-09-12 01:22:56 +08:00
test_lazy [test] skip some not compatible models 2023-08-15 23:25:14 +08:00
test_legacy [legacy] move communication and nn to legacy and refactor logger (#4671) 2023-09-11 16:24:28 +08:00
test_moe [legacy] move engine to legacy (#4560) 2023-09-05 21:53:10 +08:00
test_optimizer
test_pipeline [legacy] move communication and nn to legacy and refactor logger (#4671) 2023-09-11 16:24:28 +08:00
test_shardformer [shardformer]fix gpt2 double head (#4663) 2023-09-11 18:35:03 +08:00
test_tensor [gemini] improve compatibility and add static placement policy (#4479) 2023-08-24 09:29:25 +08:00
test_utils [legacy] move communication and nn to legacy and refactor logger (#4671) 2023-09-11 16:24:28 +08:00
test_zero [zero]fix zero ckptIO with offload (#4529) 2023-09-01 17:41:19 +08:00
__init__.py