mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!21942 modification for MindStudio 

Merge pull request !21942 from Erpim/master
This commit is contained in:
i-robot 2021-08-19 09:18:47 +00:00 committed by Gitee
parent d5b61f23ab 163c32f55a
commit e99d814933
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py
View File

@ -21,7 +21,6 @@ from te.platform.cce_build import build_config
from topi import generic from topi import generic
from topi.cce import util from topi.cce import util
from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 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 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) te.lang.cce.cce_build_code(sch, config)
return return
from impl.bn_training_reduce import bn_training_reduce_schedule_nd
sch, tensor_list = bn_training_reduce_schedule_nd(res_list) sch, tensor_list = bn_training_reduce_schedule_nd(res_list)
with build_config: with build_config:
tvm.build(sch, tensor_list, "cce", name=kernel_name) tvm.build(sch, tensor_list, "cce", name=kernel_name)