Skip to content

Commit 89d9707

Browse files
authored
[torchlib] Make matmul trace_only (#2055)
Somehow it was left behind
1 parent 456f0ec commit 89d9707

File tree

1 file changed

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

1 file changed

+3
-3
lines changed

onnxscript/function_libs/torch_lib/ops/core.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2818,7 +2818,7 @@ def aten_div_mode_int(self: TInt, other: TInt, rounding_mode: str) -> TInt:
28182818
return op.CastLike(result, self)
28192819

28202820

2821-
@torch_op("aten::dot")
2821+
@torch_op("aten::dot", trace_only=True)
28222822
def aten_dot(self: TFloat, tensor: TFloat) -> TFloat:
28232823
"""dot(Tensor self, Tensor tensor) -> Tensor"""
28242824

@@ -5114,7 +5114,7 @@ def aten_masked_select_backward(
51145114
raise NotImplementedError()
51155115

51165116

5117-
@torch_op("aten::matmul")
5117+
@torch_op("aten::matmul", trace_only=True)
51185118
def aten_matmul(
51195119
self: TRealUnlessInt16OrInt8, other: TRealUnlessInt16OrInt8
51205120
) -> TRealUnlessInt16OrInt8:
@@ -5670,7 +5670,7 @@ def aten_multiply(self: TensorType, other: TensorType) -> TensorType:
56705670
raise NotImplementedError()
56715671

56725672

5673-
@torch_op("aten::mv")
5673+
@torch_op("aten::mv", trace_only=True)
56745674
def aten_mv(self: TensorType, vec: TensorType) -> TensorType:
56755675
"""mv(Tensor self, Tensor vec) -> Tensor"""
56765676

0 commit comments

Comments
 (0)