Fix formatting

__original_commit__ = fairinternal/xformers@17aa74f
facebookresearch · Feb 10, 2025 · 6df18d9 · 6df18d9
1 parent 144ca64
commit 6df18d9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/xformers/ops/sequence_parallel_fused_ops.py b/xformers/ops/sequence_parallel_fused_ops.py
@@ -243,6 +243,7 @@ def fused_allgather_and_anything(
 
     # Fast path
     else:
+
         def my_wrapper(t, rank):
             my_matmul(t.squeeze(0), rank, _default_stream_factory)
 
@@ -379,6 +380,7 @@ def fused_anything_and_reducescatter(
 
     # Fast path
     else:
+
         def my_wrapper(rank, t):
             my_matmul(t.squeeze(0), rank, _default_stream_factory)