Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
FlashMLA
Commits
bd0083f5
Commit
bd0083f5
authored
May 25, 2026
by
zhanghj2
Browse files
test have_topk_length performance
parent
0728420c
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
tests/test_flash_mla_sparse_prefill.py
tests/test_flash_mla_sparse_prefill.py
+2
-1
No files found.
tests/test_flash_mla_sparse_prefill.py
View file @
bd0083f5
...
...
@@ -161,10 +161,11 @@ if __name__ == '__main__':
]
performance_cases
=
[
TestParam
(
s_q
,
s_kv
,
topk
,
h_q
=
h_q
,
d_qk
=
d_qk
,
have_attn_sink
=
have_attn_sink
)
TestParam
(
s_q
,
s_kv
,
topk
,
h_q
=
h_q
,
d_qk
=
d_qk
,
have_attn_sink
=
have_attn_sink
,
have_topk_length
=
have_topk_length
)
for
(
d_qk
,
h_q
,
topk
,
s_kv_list
)
in
performance_case_templates
for
s_q
in
[
4096
]
for
have_attn_sink
in
[
False
,
True
]
for
have_topk_length
in
[
False
,
True
]
for
s_kv
in
s_kv_list
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment