From 392adb87d7dd2df7b3bf16257db361d8d7604fa8 Mon Sep 17 00:00:00 2001 From: andrewwhitecdw Date: Wed, 12 Aug 2026 15:03:04 -0500 Subject: [PATCH] fix: Optional segment position annotations in mask helpers Squashed to single commit for review. Original PR: https://github.com/andrewwhitecdw/TransformerEngine/pull/9 --- tests/jax/test_fused_attn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/jax/test_fused_attn.py b/tests/jax/test_fused_attn.py index 1dd92fe181..f4f279c868 100644 --- a/tests/jax/test_fused_attn.py +++ b/tests/jax/test_fused_attn.py @@ -481,7 +481,7 @@ def _check_configs(self): if self.attn_bias_type is not AttnBiasType.NO_BIAS: pytest.skip(f"cuDNN does not support pre or post scale bias for BRCM") if self.dropout_prob != 0.0: - pytest.skip(f"cuDNN does not support non-zero dropoouts for BRCM") + pytest.skip("cuDNN does not support non-zero dropouts for BRCM") if self.qkv_layout.is_qkvpacked(): if self.max_seqlen_q != self.max_seqlen_kv: