[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
NVIDIA · Dec 20, 2024 · 8c97ce8 · 8c97ce8
1 parent bb458ee
commit 8c97ce8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/transformer_engine/pytorch/attention.py b/transformer_engine/pytorch/attention.py
@@ -604,7 +604,8 @@ def get_attention_backend(
             use_fused_attention = False
         elif cudnn_version >= (9, 6, 0) and qkv_format == "thd":
             logger.debug(
-                "Disabling FusedAttention as it does not support context parallelism with THD for cuDNN 9.6+"
+                "Disabling FusedAttention as it does not support context parallelism with THD for"
+                " cuDNN 9.6+"
             )
             use_fused_attention = False