Browse Source

[MSLITE][Develop] fix bug of arm fp16 cpu op: reopen deconv depthwise

tags/v1.1.0
yangruoqi713 5 years ago
parent
commit
e696b6a430
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/lite/src/runtime/kernel/arm/fp16/deconvolution_depthwise_fp16.cc

+ 3
- 0
mindspore/lite/src/runtime/kernel/arm/fp16/deconvolution_depthwise_fp16.cc View File

@@ -176,6 +176,7 @@ int DeconvolutionDepthwiseFp16CPUKernel::Run() {
}

if (!need_align_) {
memset(execute_output_, 0, out_tensors_.at(kOutputIndex)->ElementsNum() * sizeof(float16_t));
packed_output_ = execute_output_;
}
ret = ParallelLaunch(this->context_->thread_pool_, DeconvDwFp16Run, this, conv_param_->thread_num_);
@@ -243,4 +244,6 @@ kernel::LiteKernel *CpuDeconvDwFp16KernelCreator(const std::vector<lite::Tensor
}
return kernel;
}

REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_DeDepthwiseConv2D, CpuDeconvDwFp16KernelCreator)
} // namespace mindspore::kernel

Loading…
Cancel
Save