[torchlib] Make matmul trace_only (#2055)

Somehow it was left behind
microsoft · Feb 12, 2025 · 89d9707 · 89d9707
1 parent 456f0ec
commit 89d9707
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py
@@ -2818,7 +2818,7 @@ def aten_div_mode_int(self: TInt, other: TInt, rounding_mode: str) -> TInt:
     return op.CastLike(result, self)
 
 
-@torch_op("aten::dot")
+@torch_op("aten::dot", trace_only=True)
 def aten_dot(self: TFloat, tensor: TFloat) -> TFloat:
     """dot(Tensor self, Tensor tensor) -> Tensor"""
 
@@ -5114,7 +5114,7 @@ def aten_masked_select_backward(
     raise NotImplementedError()
 
 
-@torch_op("aten::matmul")
+@torch_op("aten::matmul", trace_only=True)
 def aten_matmul(
     self: TRealUnlessInt16OrInt8, other: TRealUnlessInt16OrInt8
 ) -> TRealUnlessInt16OrInt8:
@@ -5670,7 +5670,7 @@ def aten_multiply(self: TensorType, other: TensorType) -> TensorType:
     raise NotImplementedError()
 
 
-@torch_op("aten::mv")
+@torch_op("aten::mv", trace_only=True)
 def aten_mv(self: TensorType, vec: TensorType) -> TensorType:
     """mv(Tensor self, Tensor vec) -> Tensor"""