From dba112fb525206f446e46e2532dedd3e888cf655 Mon Sep 17 00:00:00 2001 From: kistenklaus Date: Thu, 4 Sep 2025 23:01:45 +0200 Subject: [PATCH] fix _process_padding bug (#2539) --- onnxscript/function_libs/torch_lib/ops/nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxscript/function_libs/torch_lib/ops/nn.py b/onnxscript/function_libs/torch_lib/ops/nn.py index bccddb88a6..88b5bf807e 100644 --- a/onnxscript/function_libs/torch_lib/ops/nn.py +++ b/onnxscript/function_libs/torch_lib/ops/nn.py @@ -1503,7 +1503,7 @@ def _process_padding(padding: Sequence[INT64 | int], rank: int) -> INT64: paddings = [*paddings, *zeros] # Interleave the padding values paddings = paddings[-2::-2] + paddings[-1::-2] - return op.Concat(paddings, axis=0) + return op.Concat(*paddings, axis=0) @torch_op("aten::pad", trace_only=True)