Skip to content

Commit 6a790cf

Browse files
Fix the skip message
Signed-off-by: Kshitij Lakhani <[email protected]>
1 parent 4a2a993 commit 6a790cf

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/jax/test_fused_attn.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -355,8 +355,7 @@ def _check_configs(self):
355355
and self.attn_bias_type is not AttnBiasType.NO_BIAS
356356
):
357357
pytest.skip(
358-
"For Blackwell, there is no bprop kernel support for dropout + deterministic (bias)"
359-
" config "
358+
"For sm100, bprop kernel support for dropout + determinism (bias) is not supported"
360359
)
361360
# Test the MLA case where head dims for qk differ from head dims for v, only if the tensors
362361
# are provided in BSHD_BSHD_BSHD or THD_THD_THD formats

0 commit comments

Comments
 (0)