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

Merge pull request !5221 from sunsuodong/fix_matmul_create
pull/5221/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit cfd0b5f48a

@ -35,30 +35,15 @@ kernel::LiteKernel *CpuMatmulKernelCreator(const std::vector<lite::tensor::Tenso
auto input_tensor = inputs.at(kInputIndex); auto input_tensor = inputs.at(kInputIndex);
auto data_type = input_tensor->data_type(); auto data_type = input_tensor->data_type();
kernel::LiteKernel *kernel = nullptr; kernel::LiteKernel *kernel = nullptr;
switch (data_type) { if (data_type == kNumberTypeInt8 || data_type == kNumberTypeUInt8) {
case kNumberTypeInt8: kernel = new (std::nothrow) MatmulInt8CPUKernel(opParameter, inputs, outputs, ctx, primitive);
case kNumberTypeUInt8: { } else {
kernel = new (std::nothrow) MatmulInt8CPUKernel(opParameter, inputs, outputs, ctx, primitive); kernel = new (std::nothrow) MatmulCPUKernel(opParameter, inputs, outputs, ctx, primitive);
if (kernel == nullptr) { }
MS_LOG(ERROR) << "kernel is nullptr."; if (kernel == nullptr) {
return 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;
} }
auto ret = kernel->Init(); auto ret = kernel->Init();
if (ret != RET_OK) { if (ret != RET_OK) {
delete kernel; delete kernel;

Loading…
Cancel
Save