Skip to content

Commit b3ba23b

Browse files
authored
test-perf (#55)
stack-info: PR: #55, branch: drisspg/stack/14
1 parent e5699dd commit b3ba23b

2 files changed

Lines changed: 523 additions & 0 deletions

File tree

benchmarks/configs/base_fa4.yaml

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
dtype: float16
2+
batch_sizes:
3+
- 2
4+
- 4
5+
seqlens:
6+
- 1024
7+
- 2048
8+
- 8192
9+
- 16384
10+
num_heads:
11+
- 16
12+
head_dims:
13+
- 128
14+
causal: true
15+
include_non_causal: True
16+
save_path: null

0 commit comments

Comments
 (0)