diff --git a/mindspore/lite/src/scheduler.cc b/mindspore/lite/src/scheduler.cc index 6d0d979873..cd73d18aef 100644 --- a/mindspore/lite/src/scheduler.cc +++ b/mindspore/lite/src/scheduler.cc @@ -209,6 +209,9 @@ kernel::LiteKernel *Scheduler::FindBackendKernel(const std::vector &in #endif #if SUPPORT_NPU if (context_->IsNpuEnabled()) { + if (desc.data_type == kNumberTypeFloat16) { + desc.data_type = kNumberTypeFloat32; + } kernel::KernelKey npu_desc{kNPU, desc.data_type, desc.type}; auto *kernel = KernelRegistry::GetInstance()->GetKernel(in_tensors, out_tensors, primitive, context_, npu_desc); if (kernel != nullptr) {