From 2f291b479e6e9bf6f600e5180e741ed59bd8c317 Mon Sep 17 00:00:00 2001 From: Yuan Yao Date: Fri, 7 Nov 2025 20:10:50 -0800 Subject: [PATCH 1/2] add missing output_size kwarg to repeat_interleave --- onnxscript/function_libs/torch_lib/ops/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py index 2cbecdcfc..6f15d5aa0 100644 --- a/onnxscript/function_libs/torch_lib/ops/core.py +++ b/onnxscript/function_libs/torch_lib/ops/core.py @@ -7538,7 +7538,7 @@ def aten_repeat(self: TTensor, repeats: Sequence[TInt]) -> TTensor: @torch_op("aten::repeat_interleave.self_int", trace_only=True) def aten_repeat_interleave_self_int( - self: TensorType, repeats: int, dim: Optional[int] = None + self: TensorType, repeats: int, dim: Optional[int] = None, output_size: Optional[int] = None ) -> TensorType: """repeat_interleave.self_int(Tensor self, SymInt repeats, int? dim=None, *, SymInt? output_size=None) -> Tensor From c5dcce4fa01e597b0d773ae4b6afe19a218aad92 Mon Sep 17 00:00:00 2001 From: Yuan Yao Date: Mon, 10 Nov 2025 10:31:43 -0800 Subject: [PATCH 2/2] fix lint Signed-off-by: Yuan Yao --- onnxscript/function_libs/torch_lib/ops/core.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py index 6f15d5aa0..1a48a528c 100644 --- a/onnxscript/function_libs/torch_lib/ops/core.py +++ b/onnxscript/function_libs/torch_lib/ops/core.py @@ -7538,7 +7538,10 @@ def aten_repeat(self: TTensor, repeats: Sequence[TInt]) -> TTensor: @torch_op("aten::repeat_interleave.self_int", trace_only=True) def aten_repeat_interleave_self_int( - self: TensorType, repeats: int, dim: Optional[int] = None, output_size: Optional[int] = None + self: TensorType, + repeats: int, + dim: Optional[int] = None, + output_size: Optional[int] = None, ) -> TensorType: """repeat_interleave.self_int(Tensor self, SymInt repeats, int? dim=None, *, SymInt? output_size=None) -> Tensor