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

for more information, see https://pre-commit.ci
NVIDIA · Sep 17, 2024 · 6a86660 · 6a86660
1 parent 1db61e2
commit 6a86660
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/transformer_engine/pytorch/attention.py b/transformer_engine/pytorch/attention.py
@@ -5110,10 +5110,10 @@ def convert_to_torch_float8(tensor, dtype):
             if fp8 and fp8_meta["recipe"].fp8_mha:
                 output = Float8Tensor.make_like(
                     output,
-                    data=output._data.reshape(
-                        batch_size, max_seqlen_q // cp_size, -1
-                    ).transpose(0,1).contiguous()
-                    )
+                    data=output._data.reshape(batch_size, max_seqlen_q // cp_size, -1)
+                    .transpose(0, 1)
+                    .contiguous(),
+                )
             else:
                 output = (
                     output.view(batch_size, max_seqlen_q // cp_size, -1)