You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests
Yuanheng Zhao a37f82629d
[Inference/SpecDec] Add Speculative Decoding Implementation (#5423)
8 months ago
..
kit [devops] remove post commit ci (#5566) 8 months ago
test_analyzer
test_auto_parallel [npu] change device to accelerator api (#5239) 11 months ago
test_autochunk
test_booster [shardformer] fix pipeline forward error if custom layer distribution is used (#5189) 8 months ago
test_checkpoint_io [devops] remove post commit ci (#5566) 8 months ago
test_cluster [shardformer] Sequence Parallelism Optimization (#5533) 8 months ago
test_config
test_device
test_fx
test_gptq [devops] remove post commit ci (#5566) 8 months ago
test_infer [Inference/SpecDec] Add Speculative Decoding Implementation (#5423) 8 months ago
test_lazy [devops] remove post commit ci (#5566) 8 months ago
test_legacy [npu] change device to accelerator api (#5239) 11 months ago
test_moe [hotfix] set return_outputs=False in examples and polish code (#5404) 8 months ago
test_optimizer [devops] remove post commit ci (#5566) 8 months ago
test_pipeline [devops] remove post commit ci (#5566) 8 months ago
test_shardformer [shardformer] Sequence Parallelism Optimization (#5533) 8 months ago
test_smoothquant
test_tensor fixed layout converter caching and updated tester 8 months ago
test_zero [npu] change device to accelerator api (#5239) 11 months ago
__init__.py