From d2783979d3b0ceec872e829297f2901c44514dc8 Mon Sep 17 00:00:00 2001 From: sunsuodong Date: Wed, 26 Aug 2020 12:14:47 +0800 Subject: [PATCH] fix_matmul_create --- .../runtime/kernel/arm/base/matmul_base.cc | 31 +++++-------------- 1 file changed, 8 insertions(+), 23 deletions(-) diff --git a/mindspore/lite/src/runtime/kernel/arm/base/matmul_base.cc b/mindspore/lite/src/runtime/kernel/arm/base/matmul_base.cc index 4d73c7c658..01b719e1b1 100644 --- a/mindspore/lite/src/runtime/kernel/arm/base/matmul_base.cc +++ b/mindspore/lite/src/runtime/kernel/arm/base/matmul_base.cc @@ -35,30 +35,15 @@ kernel::LiteKernel *CpuMatmulKernelCreator(const std::vectordata_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;