mirror of https://github.com/hpcaitech/ColossalAI
![]() * clean requirements * modify example inference struct * add test ci scripts * mark test_infer as submodule * rm deprecated cls & deps * import of HAS_FLASH_ATTN * prune inference tests to be run * prune triton kernel tests * increment pytest timeout mins * revert import path in openmoe |
||
---|---|---|
.. | ||
__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 |