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
vllm_cscc
Commits
7e7db043
Unverified
Commit
7e7db043
authored
Sep 09, 2025
by
Nick Hill
Committed by
GitHub
Sep 09, 2025
Browse files
[CI] Retry flaky fp8 cutlass mla tests (#24536)
Signed-off-by:
Nick Hill
<
nhill@redhat.com
>
parent
41f160b9
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
1 deletion
+7
-1
tests/kernels/test_cutlass_mla_decode.py
tests/kernels/test_cutlass_mla_decode.py
+7
-1
No files found.
tests/kernels/test_cutlass_mla_decode.py
View file @
7e7db043
...
...
@@ -49,7 +49,13 @@ CUTLASS_MLA_UNSUPPORTED_REASON = \
@
pytest
.
mark
.
parametrize
(
"block_size"
,
[
64
])
@
pytest
.
mark
.
parametrize
(
"causal"
,
[
True
])
@
pytest
.
mark
.
parametrize
(
"varlen"
,
[
False
,
True
])
@
pytest
.
mark
.
parametrize
(
"torch_dtype"
,
[
torch
.
bfloat16
,
torch
.
float8_e4m3fn
])
@
pytest
.
mark
.
parametrize
(
"torch_dtype"
,
[
torch
.
bfloat16
,
# fp8 can have occasional precision-related failures.
pytest
.
param
(
torch
.
float8_e4m3fn
,
marks
=
pytest
.
mark
.
flaky
(
reruns
=
2
))
])
@
torch
.
inference_mode
()
def
test_cutlass_mla_decode
(
b
,
s_q
,
mean_sk
,
h_q
,
h_kv
,
d
,
dv
,
block_size
,
causal
,
varlen
,
torch_dtype
):
...
...
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