Browse Source

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

GitOrigin-RevId: 61ef7a50ff
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
585aa561f3
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      imperative/python/megengine/functional/math.py

+ 3
- 3
imperative/python/megengine/functional/math.py View File

@@ -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(
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(
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)


Loading…
Cancel
Save