github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-04 09:52:50 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
0aa27f196109bfb4ce6171d7ce921052b9eee969
ColossalAI/tests/test_infer/test_ops
History
yuehuayingxueluo 0aa27f1961 [Inference]Move benchmark-related code to the example directory. (#5408) 
* move benchmark-related code to the example directory.

* fix bugs in test_fused_rotary_embedding.py
2024-02-28 16:46:03 +08:00
..
cuda
[Inference]Add CUDA KVCache Kernel (#5406)
2024-02-28 14:36:50 +08:00
triton
[Inference]Move benchmark-related code to the example directory. (#5408)
2024-02-28 16:46:03 +08:00
__init__.py
[Inference]Add CUDA KVCache Kernel (#5406)
2024-02-28 14:36:50 +08:00