xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!24067 landmar_detector model bugfix 

Merge pull request !24067 from Greatpan/master
This commit is contained in:
i-robot 2021-09-26 03:35:00 +00:00 committed by Gitee
parent 9eebaefa2c fb5e77805e
commit d23a5f237d
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

@ -1469,7 +1469,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;