Update core.py

microsoft · Jul 9, 2024 · 931059f · 931059f
1 parent c5ea84a
commit 931059f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py
@@ -6583,10 +6583,11 @@ def aten_prelu_backward(
     raise NotImplementedError()
 
 
-@torch_op(("aten::prod", "aten::prod.dim_int"), trace_only=True)
+@torch_op(("aten::prod.dim_int"), trace_only=True)
 def aten_prod(self: TReal, dim: int, keepdim: bool = False) -> TReal:
     """prod(Tensor self, *, ScalarType? dtype=None) -> Tensor"""
 
+    # Todo: add test for this function later
     return op.ReduceProd(self, axes=[dim], keepdims=keepdim)