提交 585aa561 编写于 作者: M Megvii Engine Team

fix(functional): replace usage of deprecated get_conv_execution_strategy()

GitOrigin-RevId: 61ef7a50ff9560acdc332aad04cfeafd5baf1827
上级 dcfb6a53
......@@ -16,7 +16,7 @@ from ..core.ops import builtin
from ..core.ops.special import Const
from ..core.tensor import utils
from ..tensor import Tensor
from .debug_param import get_conv_execution_strategy
from .debug_param import get_execution_strategy
from .elemwise import clip, exp, log, log1p
from .tensor import broadcast_to, concat, expand_dims, reshape, squeeze
......@@ -864,7 +864,7 @@ def matmul(
transposeB=transpose_b,
compute_mode=compute_mode,
format=format,
strategy=get_conv_execution_strategy(),
strategy=get_execution_strategy(),
)
else:
op = builtin.MatrixMul(
......@@ -872,7 +872,7 @@ def matmul(
transposeB=transpose_b,
compute_mode=compute_mode,
format=format,
strategy=get_conv_execution_strategy(),
strategy=get_execution_strategy(),
)
(result,) = apply(op, inp1, inp2)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册