Browse Source

!5221 [MS][LITE] 20200822 ml_face_3d model fp16 version exec fail

Merge pull request !5221 from sunsuodong/fix_matmul_create
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
cfd0b5f48a
1 changed files with 8 additions and 23 deletions
  1. +8
    -23
      mindspore/lite/src/runtime/kernel/arm/base/matmul_base.cc

+ 8
- 23
mindspore/lite/src/runtime/kernel/arm/base/matmul_base.cc View File

@@ -35,30 +35,15 @@ kernel::LiteKernel *CpuMatmulKernelCreator(const std::vector<lite::tensor::Tenso
auto input_tensor = inputs.at(kInputIndex);
auto data_type = input_tensor->data_type();
kernel::LiteKernel *kernel = nullptr;
switch (data_type) {
case kNumberTypeInt8:
case kNumberTypeUInt8: {
kernel = new (std::nothrow) MatmulInt8CPUKernel(opParameter, inputs, outputs, ctx, primitive);
if (kernel == nullptr) {
MS_LOG(ERROR) << "kernel is nullptr.";
return nullptr;
}
break;
}

case kNumberTypeFloat32: {
kernel = new (std::nothrow) MatmulCPUKernel(opParameter, inputs, outputs, ctx, primitive);
if (kernel == nullptr) {
MS_LOG(ERROR) << "kernel is nullptr.";
return nullptr;
}
break;
}

default:
break;
if (data_type == kNumberTypeInt8 || data_type == kNumberTypeUInt8) {
kernel = new (std::nothrow) MatmulInt8CPUKernel(opParameter, inputs, outputs, ctx, primitive);
} else {
kernel = new (std::nothrow) MatmulCPUKernel(opParameter, inputs, outputs, ctx, primitive);
}
if (kernel == nullptr) {
MS_LOG(ERROR) << "kernel is nullptr.";
return nullptr;
}

auto ret = kernel->Init();
if (ret != RET_OK) {
delete kernel;


Loading…
Cancel
Save