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/colossalai/inference/kv_cache
Yuanheng Zhao 5f98a9d68a
[Infer] Optimize Blocked KVCache And Kernels Using It (#5325)
10 months ago
..
__init__.py [Inference] Add CacheBlock and KV-Cache Manager (#5156) 11 months ago
block_cache.py [Inference] Add CacheBlock and KV-Cache Manager (#5156) 11 months ago
kvcache_manager.py [Infer] Optimize Blocked KVCache And Kernels Using It (#5325) 10 months ago