Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

landmar_detector model bugfix

This commit is contained in:
greatpanc 2021-09-24 16:23:21 +08:00
parent 7cde7731b0
commit fb5e77805e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/src/scheduler.cc
View File

@ -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;