github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-28 13:05:26 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
6fb4bcbb2420b9f977ab74de60c6d311b6c9ed9a
ColossalAI/colossalai/kernel
History
yuehuayingxueluo 6fb4bcbb24 [Inference/opt] Fused KVCahce Memcopy (#5374) 
* fused kv memcopy

* add TODO in test_kvcache_copy.py
2024-02-07 17:15:42 +08:00
..
jit
[npu] change device to accelerator api (#5239)
2024-01-09 10:20:05 +08:00
triton
[Inference/opt] Fused KVCahce Memcopy (#5374)
2024-02-07 17:15:42 +08:00
__init__.py
[feat] refactored extension module (#5298)
2024-01-25 17:01:48 +08:00
extensions
[feat] refactored extension module (#5298)
2024-01-25 17:01:48 +08:00
kernel_loader.py
[feat] refactored extension module (#5298)
2024-01-25 17:01:48 +08:00