Browse Source

fix ml_2020_ocr_cn fp16 segmenttation one thread

tags/v1.1.0
zhaodezan 5 years ago
parent
commit
cc0a742e16
2 changed files with 27 additions and 0 deletions
  1. +26
    -0
      mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.cc
  2. +1
    -0
      mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.h

+ 26
- 0
mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.cc View File

@@ -21,6 +21,7 @@
#include "schema/model_generated.h"
#include "src/kernel_registry.h"
#include "src/runtime/runtime_api.h"
#include "src/ops/populate/populate_register.h"
#include "include/errorcode.h"

using mindspore::kernel::KERNEL_ARCH::kCPU;
@@ -97,6 +98,31 @@ int ArithmeticFP16CPUKernel::Init() {
return ReSize();
}

int ArithmeticFP16CPUKernel::PreProcess() {
if (!InferShapeDone()) {
(const_cast<mindspore::lite::PrimitiveC *>(primitive_))->SetInferFlag(true);
auto ret = (const_cast<mindspore::lite::PrimitiveC *>(primitive_))->InferShape(in_tensors_, out_tensors_);
if (ret != 0) {
(const_cast<mindspore::lite::PrimitiveC *>(primitive_))->SetInferFlag(false);
MS_LOG(ERROR) << "InferShape fail!";
return ret;
}
param_ = reinterpret_cast<ArithmeticParameter *>(PopulateArithmetic(primitive_));
ret = ReSize();
if (ret != 0) {
MS_LOG(ERROR) << "ReSize fail!ret: " << ret;
return ret;
}
}

auto outputs = this->out_tensors();
for (auto *output : outputs) {
MS_ASSERT(output != nullptr);
output->MallocData();
}
return RET_OK;
}

int ArithmeticFP16CPUKernel::ReSize() {
param_->in_elements_num0_ = in_tensors_[0]->ElementsNum();
param_->in_elements_num1_ = in_tensors_[1]->ElementsNum();


+ 1
- 0
mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.h View File

@@ -44,6 +44,7 @@ class ArithmeticFP16CPUKernel : public LiteKernel {
~ArithmeticFP16CPUKernel() = default;

int Init() override;
int PreProcess() override;
int ReSize() override;
int Run() override;
int DoArithmetic(int task_id);


Loading…
Cancel
Save