Browse Source

correct data type before schedule npu op

tags/v1.2.0-rc1
zhaozhenlong 4 years ago
parent
commit
09c338e092
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/lite/src/scheduler.cc

+ 3
- 0
mindspore/lite/src/scheduler.cc View File

@@ -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) {


Loading…
Cancel
Save