From d62e3c7445bd5ba246846e0df5ea2a23cef1c6e2 Mon Sep 17 00:00:00 2001 From: andrewwhitecdw Date: Wed, 12 Aug 2026 15:03:45 -0500 Subject: [PATCH] fix: Unreachable backend check after earlier backend skip Squashed to single commit for review. Original PR: https://github.com/andrewwhitecdw/TransformerEngine/pull/10 --- tests/jax/test_fused_attn.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/tests/jax/test_fused_attn.py b/tests/jax/test_fused_attn.py index 1dd92fe181..0eb5f0d9ae 100644 --- a/tests/jax/test_fused_attn.py +++ b/tests/jax/test_fused_attn.py @@ -599,11 +599,6 @@ def _check_configs(self): pytest.skip( "B1SS, BHSS and 11SS bias shapes are only supported for non-padding mask" ) - elif self.backend != NVTE_Fused_Attn_Backend.NVTE_F16_arbitrary_seqlen: - pytest.skip( - "B1SS, BHSS and 11SS bias shapes are only supported for " - "the F16_arbitrary_seqlen backend." - ) def _setup_inputs(self): self._check_configs()