[torchlib] Support mod and eq on SymInt (#1879)

Add some missing operators from pytorch/pytorch#136524 /cc @justinchuby
microsoft · Sep 24, 2024 · 500ed62 · 500ed62
1 parent 99ae64e
commit 500ed62
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py
@@ -3269,7 +3269,7 @@ def aten_empty_strided(
  return op.Expand(zero, size)
 
 
-@torch_op(("aten::eq", "aten::eq.Tensor", "aten::eq.Scalar"))
+@torch_op(("aten::eq", "aten::eq.Tensor", "aten::eq.Scalar", "_operator::eq"))
 def aten_eq(self: TTensor, other: TTensor) -> BOOL:
  """eq.Tensor(Tensor self, Tensor other) -> Tensor"""
 
@@ -7085,7 +7085,7 @@ def aten_remainder(self: TFloatOrBFloat16, other: TFloatOrBFloat16) -> TFloatOrB
  return op.Sub(self, op.Mul(rounded_quotient, other))
 
 
-@torch_op(("aten::remainder.Tensor", "aten::remainder.Scalar"))
+@torch_op(("aten::remainder.Tensor", "aten::remainder.Scalar", "_operator::mod"))
 def aten_remainder_int(self: TInt, other: TInt) -> TInt:
  """remainder.Tensor(Tensor self, Tensor other) -> Tensor"""