Fix lint: wrap long function call arguments

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Signed-off-by: Justin Chu <justinchu@microsoft.com>

Author: justinchuby

src/mobius/rewrite_rules/_group_query_attention.py

@@ -647,7 +647,13 @@ def rewrite(
             gqa_attrs["softcap"] = softcap
 
         outputs = op.GroupQueryAttention(
-            q, k, v, past_key, past_value, self._seqlens_k, self._total_seq_len,
+            q,
+            k,
+            v,
+            past_key,
+            past_value,
+            self._seqlens_k,
+            self._total_seq_len,
             _domain="com.microsoft",
             _outputs=3,
             **gqa_attrs,

src/mobius/rewrite_rules/_separate_rope.py

@@ -134,7 +134,10 @@ def rewrite(self, op, q, k, v, gqa_out, present_key, present_value, **_):
         remaining = list(gqa_node.inputs[3:7])
 
         outputs = op.GroupQueryAttention(
-            q_rot, k_rot, v, *remaining,
+            q_rot,
+            k_rot,
+            v,
+            *remaining,
             _domain="com.microsoft",
             _outputs=3,
             **attrs,

src/mobius/rewrite_rules/_unpack_qkv.py

@@ -286,7 +286,10 @@ def _proj(w: np.ndarray, name: str) -> ir.Value:
         remaining = list(gqa_node.inputs[3:])  # everything after the packed slot
 
         outputs = op.GroupQueryAttention(
-            q, k, v, *remaining,
+            q,
+            k,
+            v,
+            *remaining,
             _domain="com.microsoft",
             _outputs=3,
             **attrs,