Skip to content

Commit 471d6da

Browse files
committed
style: format fused attention sanity skip
1 parent 9581773 commit 471d6da

1 file changed

Lines changed: 1 addition & 6 deletions

File tree

tests/pytorch/test_sanity.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -806,12 +806,7 @@ def test_sanity_T5_126m():
806806
def test_sanity_amp_and_nvfuser(dtype, fp8_recipe, model, skip_wgrad):
807807
config = model_configs[model]
808808

809-
if (
810-
dtype == torch.float16
811-
and fp8_recipe is None
812-
and model == "small"
813-
and skip_wgrad
814-
):
809+
if dtype == torch.float16 and fp8_recipe is None and model == "small" and skip_wgrad:
815810
pytest.skip("Known fused_attn_bwd segfault for this AMP combo")
816811

817812
if fp8_recipe is not None:

0 commit comments

Comments
 (0)