correct data type before schedule npu op

4 years ago · 09c338e092
parent adfe6e1bc2
commit 09c338e092
1 changed files with 3 additions and 0 deletions
--- a/mindspore/lite/src/scheduler.cc
+++ b/mindspore/lite/src/scheduler.cc
@ -209,6 +209,9 @@ kernel::LiteKernel *Scheduler::FindBackendKernel(const std::vector<Tensor *> &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) {