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

!27115 [MS][LITE] modify custom kernel subgraph type 

Merge pull request !27115 from jianghui58/subgraph_mas
This commit is contained in:
i-robot 2021-12-03 01:54:32 +00:00 committed by Gitee
parent 204822ecd8 e2a2bcd4e7
commit ba803c1d16
1 changed files with 2 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1119,15 +1119,6 @@ kernel::SubGraphType GetKernelSubGraphType(const kernel::LiteKernel *kernel, con
}
auto desc = kernel->desc();
if (desc.provider != kernel::kBuiltin) {
if (desc.arch == kernel::KERNEL_ARCH::kGPU) {
if (desc.data_type == kNumberTypeFloat16) {
return kernel::kGpuFp16SubGraph;
}
return kernel::kGpuFp32SubGraph;
}
return kernel::kCustomSubGraph;
}
if (desc.arch == kernel::KERNEL_ARCH::kGPU) {
if (desc.data_type == kNumberTypeFloat16) {
return kernel::kGpuFp16SubGraph;
@ -1152,6 +1143,8 @@ kernel::SubGraphType GetKernelSubGraphType(const kernel::LiteKernel *kernel, con
return kernel::kCpuFP32SubGraph;
}
}
} else if (desc.arch == kernel::KERNEL_ARCH::kCustom) {
return kernel::kCustomSubGraph;
}
return kernel::kNotSubGraph;
}