Run linter

pytorch · May 7, 2024 · b15097e · b15097e
1 parent 546ca09
commit b15097e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_xla/experimental/megablox/gmm.py b/torch_xla/experimental/megablox/gmm.py
@@ -381,5 +381,5 @@ def gmm(
 ):
   payload = _trace_kernel_payload(lhs, rhs, group_sizes)
   out = _gmm(lhs, rhs, group_sizes, payload, preferred_element_type, tiling,
-              group_offset, existing_out, transpose_rhs, interpret)
-  return out
+             group_offset, existing_out, transpose_rhs, interpret)
+  return out