We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4a2a993 commit 6a790cfCopy full SHA for 6a790cf
tests/jax/test_fused_attn.py
@@ -355,8 +355,7 @@ def _check_configs(self):
355
and self.attn_bias_type is not AttnBiasType.NO_BIAS
356
):
357
pytest.skip(
358
- "For Blackwell, there is no bprop kernel support for dropout + deterministic (bias)"
359
- " config "
+ "For sm100, bprop kernel support for dropout + determinism (bias) is not supported"
360
)
361
# Test the MLA case where head dims for qk differ from head dims for v, only if the tensors
362
# are provided in BSHD_BSHD_BSHD or THD_THD_THD formats
0 commit comments