From 585aa561f369e2f3fcdaa30e4b52e375bf4cdb00 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 11 Mar 2021 13:41:33 +0800 Subject: [PATCH] fix(functional): replace usage of deprecated get_conv_execution_strategy() GitOrigin-RevId: 61ef7a50ff9560acdc332aad04cfeafd5baf1827 --- imperative/python/megengine/functional/math.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/imperative/python/megengine/functional/math.py b/imperative/python/megengine/functional/math.py index 5941db2e..63743435 100644 --- a/imperative/python/megengine/functional/math.py +++ b/imperative/python/megengine/functional/math.py @@ -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) -- GitLab