Skip to content

Commit fd257cd

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 6aa9f6b commit fd257cd

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/pytorch/attention/test_attention_with_cp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -390,7 +390,7 @@ def test_cp_with_fused_attention(
390390
if fused_attn_supported and config.attn_mask_type in ["causal", "padding_causal"]:
391391
config_copy = copy.deepcopy(config)
392392
config_copy.context_parallel = False
393-
config_copy.attn_mask_type = config.attn_mask_type+"_bottom_right"
393+
config_copy.attn_mask_type = config.attn_mask_type + "_bottom_right"
394394
available_backends, _, fused_attn_backends = get_available_attention_backends(
395395
config_copy,
396396
qkv_dtype=dtypes[dtype] if dtype != "fp8" else torch.float8_e4m3fn,

0 commit comments

Comments
 (0)