mirror of https://github.com/hpcaitech/ColossalAI
![]() * [release] update version * [devops] update compatibility test * [devops] update compatibility test * [devops] update compatibility test * [devops] update compatibility test * [test] fix ddp plugin test * [test] fix gptj and rpc test * [devops] fix cuda ext compatibility * [inference] fix flash decoding test * [inference] fix flash decoding test |
||
---|---|---|
.. | ||
__init__.py | ||
kernel_utils.py | ||
test_context_attn_unpad.py | ||
test_decoding_attn.py | ||
test_fused_rotary_embedding.py | ||
test_kvcache_copy.py | ||
test_rmsnorm_triton.py | ||
test_rotary_embdding_unpad.py | ||
test_xine_copy.py |