From fb5e77805e2439df19e8857c7b9c4c8e60b50fa4 Mon Sep 17 00:00:00 2001 From: greatpanc Date: Fri, 24 Sep 2021 16:23:21 +0800 Subject: [PATCH] landmar_detector model bugfix --- mindspore/lite/src/scheduler.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/lite/src/scheduler.cc b/mindspore/lite/src/scheduler.cc index d316ed834cb..c7ebebd3c20 100644 --- a/mindspore/lite/src/scheduler.cc +++ b/mindspore/lite/src/scheduler.cc @@ -1467,7 +1467,7 @@ bool KernelFitCurrentSubGraph(const kernel::SubGraphType subgraph_type, const ke if (kernel.desc().arch != kernel::KERNEL_ARCH::kGPU) { return false; } - return (kernel.desc().data_type == kNumberTypeFloat16); + return (kernel.desc().data_type != kNumberTypeFloat32); case kernel::SubGraphType::kGpuFp32SubGraph: if (kernel.desc().arch != kernel::KERNEL_ARCH::kGPU) { return false;