diff --git a/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/array_reduce_gpu_kernel.cc b/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/array_reduce_gpu_kernel.cc index 030a923cf9b..39de2485fe3 100644 --- a/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/array_reduce_gpu_kernel.cc +++ b/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/array_reduce_gpu_kernel.cc @@ -179,6 +179,8 @@ void ArrayReduceGpuKernelMod::InitCudnnResource() { int ArrayReduceGpuKernelMod::Resize(const BaseOperatorPtr &base_operator, const std::vector &inputs, const std::vector &outputs, const std::map &inputsOnHost) { + need_skip_execute_ = false; + all_match_ = false; int ret = KernelMod::Resize(base_operator, inputs, outputs, inputsOnHost); if (ret != KRET_OK) { InitCudnnResource();