This website requires JavaScript.
Explore
Help
Register
Sign In
github
/
ColossalAI
Watch
1
Star
0
Fork
0
You've already forked ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI.git
synced
2026-05-01 20:33:02 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
db7b3051f4379862f88790bf1653ddb6443c002e
ColossalAI
/
examples
/
inference
/
benchmark_ops
History
Yuanheng Zhao
8754abae24
[Fix] Fix & Update Inference Tests (compatibility w/ main)
2024-05-05 16:28:56 +00:00
..
benchmark_context_attn_unpad.py
[Fix] Fix & Update Inference Tests (compatibility w/ main)
2024-05-05 16:28:56 +00:00
benchmark_decoding_attn.py
[Fix] Fix & Update Inference Tests (compatibility w/ main)
2024-05-05 16:28:56 +00:00
benchmark_flash_decoding_attention.py
[Fix] Fix & Update Inference Tests (compatibility w/ main)
2024-05-05 16:28:56 +00:00
benchmark_fused_rotary_embdding_unpad.py
[Fix] Fix & Update Inference Tests (compatibility w/ main)
2024-05-05 16:28:56 +00:00
benchmark_kv_cache_memcopy.py
[Fix] Fix & Update Inference Tests (compatibility w/ main)
2024-05-05 16:28:56 +00:00
benchmark_rmsnorm.py
feat baichuan2 rmsnorm whose hidden size equals to 5120 (
#5611
)
2024-04-19 15:34:53 +08:00
benchmark_rotary_embedding.py
[Inference/kernel]Add Fused Rotary Embedding and KVCache Memcopy CUDA Kernel (
#5418
)
2024-03-13 17:20:03 +08:00
benchmark_xine_copy.py
[Fix] Fix & Update Inference Tests (compatibility w/ main)
2024-05-05 16:28:56 +00:00
