This reverts commit 7bfe292.

onnxscript/_internal/param_manipulation.py

@@ -131,3 +131,18 @@ def tag_arguments_with_param_schemas(
             raise TypeError(f"Required input/attribute '{param}' was not provided")
 
     return tagged_args, tagged_kwargs
+
+
+def return_to_args_order(
+    param_schemas: Sequence[values.ParamSchema],
+    inputs: list[Any],
+    attributes: dict[str, Any],
+) -> list[Sequence[Any]]:
+    """Return the inputs and attributes to the order of the function signature."""
+    args = []
+    for param in param_schemas:
+        if param.name in attributes:
+            args.append(attributes.pop(param.name))
+        else:
+            args.append(inputs.pop(0))
+    return args

onnxscript/function_libs/torch_lib/graph_building/_graph_building_torch.py

@@ -422,6 +422,8 @@ def eval_function(  # type: ignore[override]
         if function.traceable:
             inputs = self._graph.preprocess_inputs(inputs)
             inputs = _wrap_torch_value_to_tensor(inputs)  # type: ignore[assignment]
+            # return to args order, as it's traced onnx function
+            args = param_manipulation.return_to_args_order(param_schemas, inputs, attributes)
             # Trace the function call instead of adding the function as a node
             return function.function(*args)
         return self._graph.add_function_call(function, inputs, attributes)