ColossalAI/colossalai/kernel
Runyu Lu b2c0d9ff2b [fix] multi graphs capture error 2024-03-11 10:49:31 +08:00
..
jit [npu] change device to accelerator api (#5239) 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 (#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 [Inference]Add CUDA KVCache Kernel (#5406) 2024-02-28 14:36:50 +08:00