Skip to content

Commit 3778f7d

Browse files
authored
Apply suggestion from @justinchuby
1 parent 2ae9f21 commit 3778f7d

File tree

1 file changed

+1
-1
lines changed
  • onnxscript/function_libs/torch_lib/ops

1 file changed

+1
-1
lines changed

onnxscript/function_libs/torch_lib/ops/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7689,7 +7689,7 @@ def aten_remainder(self: TTensor, other: float) -> TTensor:
76897689

76907690

76917691
@torch_op("aten::remainder.Scalar_Tensor", trace_only=True)
7692-
def aten_remainder_scalar_tensor(self: TTensor, other: TTensor) -> TTensor:
7692+
def aten_remainder_scalar_tensor(self: float, other: TTensor) -> TTensor:
76937693
"""remainder.Scalar_Tensor(Scalar self, Tensor other) -> Tensor"""
76947694

76957695
self_tensor = ir.tensor(self, dtype=other.dtype)

0 commit comments

Comments
 (0)