ColossalAI/colossalai/kernel at 934e31afb22d2a281464aebde074eb2f238fb812 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-25 09:12:10 +00:00

Files

History

Runyu Lu b2c0d9ff2b [fix] multi graphs capture error

2024-03-11 10:49:31 +08:00

..

[npu] change device to accelerator api (#5239 )

2024-01-09 10:20:05 +08:00

[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