mirror of https://github.com/hpcaitech/ColossalAI
537a3cbc4d
* kvmemcpy triton for new kcache layout * revise tests for new kcache layout * naive triton flash decoding - new kcache layout * rotary triton kernel - new kcache layout * remove redundancy - triton decoding * remove redundancy - triton kvcache copy * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> |
||
|---|---|---|
| .. | ||
| benchmark_ops | ||
| benchmark_llama.py | ||
| benchmark_llama3.py | ||
| llama_generation.py | ||
| run_benchmark.sh | ||