Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
benchmark_configs.py 82 Bytes
Newer Older
Lei Wang's avatar
[Benchmark] Add benchmark scripts for block sparse attention (#114)  
Lei Wang committed
1
2 
# BATCH, N_HEADS, SEQ_LEN, D_HEAD, TOPK, BLOCK
configs = [[4, 2, 256, 64, 2, 64]]