From 163c32f55aa3b646fbd027ebb3eb6a766df04b24 Mon Sep 17 00:00:00 2001 From: Erpim Date: Mon, 16 Aug 2021 19:52:51 +0800 Subject: [PATCH] modification for MindStudio --- .../ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py b/mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py index cc7938b9374..12dbf4bcac7 100644 --- a/mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py +++ b/mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py @@ -21,7 +21,6 @@ from te.platform.cce_build import build_config from topi import generic from topi.cce import util from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType -from impl.bn_training_reduce import bn_training_reduce_schedule_nd SHAPE_SIZE_LIMIT = 2147483648 @@ -100,7 +99,7 @@ def batchnorm_fold2_grad_reduce(dout, x, dout_reduce, dout_x_reduce, kernel_name te.lang.cce.cce_build_code(sch, config) return - + from impl.bn_training_reduce import bn_training_reduce_schedule_nd sch, tensor_list = bn_training_reduce_schedule_nd(res_list) with build_config: tvm.build(sch, tensor_list, "cce", name=kernel_name)