ColossalAI/tests/test_infer_ops
Yuanheng Zhao 0f2b46a41c
[kernel] Revise KVCache copy triton kernel API (#5273)
* [kernel/fix] revise kvcache copy kernel api

* fix benchmark
2024-01-16 14:41:02 +08:00
..
triton [kernel] Revise KVCache copy triton kernel API (#5273) 2024-01-16 14:41:02 +08:00