mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-04-28 03:43:01 +00:00
* add cuda KVCache kernel * annotation benchmark_kvcache_copy * add use cuda * fix import path * move benchmark scripts to example/ * rm benchmark codes in test_kv_cache_memcpy.py * rm redundancy codes * rm redundancy codes * pr was modified according to the review |
||
---|---|---|
.. | ||
test_models | ||
test_ops | ||
_utils.py | ||
test_batch_bucket.py | ||
test_config_and_struct.py | ||
test_inference_engine.py | ||
test_kvcache_manager.py | ||
test_request_handler.py |