modify qkv attention axis

Author: shubhambhokare1

onnxscript/rewriter/ort_fusions/attention.py

@@ -210,7 +210,7 @@ def rewrite(
         # Dh_v = self.bindings.get("Dh_v")
         # qkv_hidden_sizes = [Dh_q, Dh_k, Dh_v]
         if self._no_slice:
-            qkv_weight = op.Concat(q_mul, k_mul, v_mul, axis=0)
+            qkv_weight = op.Concat(q_mul, k_mul, v_mul, axis=1)
 
         if self._has_past:
             attention, present = op.Attention(