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 4d73c7c658afb663b6468740a0ce0839eeb5195d..01b719e1b1c9f5ae4cafcf2bfabd4b6201b93509 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;