diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py index 6a2d0cf4d..535da66d0 100644 --- a/onnxscript/function_libs/torch_lib/ops/core.py +++ b/onnxscript/function_libs/torch_lib/ops/core.py @@ -7119,7 +7119,9 @@ def aten_remainder(self: TFloat, other: TFloat) -> TFloat: return op.Sub(self, op.Mul(rounded_quotient, other)) -@torch_op(("aten::remainder.Tensor", "aten::remainder.Scalar", "_operator::mod")) +@torch_op( + ("aten::remainder.Tensor", "aten::remainder.Scalar", "_operator::mod"), traceable=True +) def aten_remainder_int(self: TInt, other: TInt) -> TInt: """remainder.Tensor(Tensor self, Tensor other) -> Tensor"""