From 2c7908a1338438940992d6852ac9dd80ef32010f Mon Sep 17 00:00:00 2001 From: xadupre Date: Tue, 10 Jun 2025 15:18:51 +0200 Subject: [PATCH] Add aten_maximum_real --- onnxscript/function_libs/torch_lib/ops/core.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py index 0544f2effb..8ed023e0f6 100644 --- a/onnxscript/function_libs/torch_lib/ops/core.py +++ b/onnxscript/function_libs/torch_lib/ops/core.py @@ -5329,6 +5329,13 @@ def aten_maximum_bool(self: BOOL, other: BOOL) -> BOOL: return op.Or(self, other) +@torch_op(("aten::maximum", "aten::max.other")) +def aten_maximum_real(self: RealType, other: RealType) -> RealType: + """maximum(Tensor self, Tensor other) -> Tensor""" + + return op.Max(self, other) + + @torch_op("aten::mean") def aten_mean(self: TReal) -> TReal: """mean(Tensor self, *, ScalarType? dtype=None) -> Tensor"""