github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-05 23:18:22 +00:00
Code Issues Packages Projects Releases Wiki Activity
ae24b4f025
ColossalAI/colossalai/kernel
History
Runyu Lu b2c0d9ff2b [fix] multi graphs capture error
2024-03-11 10:49:31 +08:00
..
jit [npu] change device to accelerator api () 2024-01-09 10:20:05 +08:00
triton [fix] multi graphs capture error 2024-03-11 10:49:31 +08:00
__init__.py [feat] refactored extension module () 2024-01-25 17:01:48 +08:00
extensions [feat] refactored extension module () 2024-01-25 17:01:48 +08:00
kernel_loader.py [Inference]Add CUDA KVCache Kernel () 2024-02-28 14:36:50 +08:00