mirror of https://github.com/hpcaitech/ColossalAI
![]() * revise rotary embedding * remove useless print * adapt * fix * add * fix * modeling * fix * fix * fix * fused kv copy * fused copy * colossalai/kernel/triton/no_pad_rotary_embedding.py * del padding llama * del |
||
---|---|---|
.. | ||
test_models | ||
test_ops/triton | ||
_utils.py | ||
test_batch_bucket.py | ||
test_config_and_struct.py | ||
test_inference_engine.py | ||
test_kvcache_manager.py | ||
test_request_handler.py |