Skip to content

Commit 057f406

Browse files
author
root
committed
Merge branch 'flash_attn_pad_bw_seqs' of https://github.com/sudhakarsingh27/TransformerEngine into flash_attn_pad_bw_seqs
2 parents 38f9d88 + 8e78753 commit 057f406

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

tests/pytorch/attention/test_attention_with_cp.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -386,9 +386,7 @@ def test_cp_with_fused_attention(
386386
"Deterministic mode does not support non-vanilla softmax with FusedAttention"
387387
)
388388
if config.attn_bias_type == "post_scale_bias" and is_training:
389-
pytest.skip(
390-
"Deterministic mode does not support post_scale_bias with requires_grad"
391-
)
389+
pytest.skip("Deterministic mode does not support post_scale_bias with requires_grad")
392390

393391
run_distributed(
394392
get_bash_arguments(

0 commit comments

Comments
 (0)