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