hyf152
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!27091 [MS][LITE] modify get custom kernel subgraph type 

Merge pull request !27091 from jianghui58/subgraph_r1.5
This commit is contained in:
i-robot 2021-12-02 08:27:51 +00:00 committed by Gitee
parent 5af1f4962b 063f3e3d26
commit 4c60bba36d
1 changed files with 2 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1181,19 +1181,6 @@ kernel::SubGraphKernel *CreateSubGraphKernel(const std::vector<kernel::LiteKerne
return sub_graph;
}
namespace {
kernel::SubGraphType GetCustomKernelSubGraphType(const kernel::LiteKernel *kernel) {
auto desc = kernel->desc();
if (desc.arch == kernel::KERNEL_ARCH::kGPU) {
if (desc.data_type == kNumberTypeFloat16) {
return kernel::kGpuFp16SubGraph;
}
return kernel::kGpuFp32SubGraph;
}
return kernel::kCustomSubGraph;
}
} // namespace
kernel::SubGraphType GetKernelSubGraphType(const kernel::LiteKernel *kernel, const InnerContext &context,
bool is_controlflow = false) {
if (kernel == nullptr) {
@ -1201,9 +1188,6 @@ kernel::SubGraphType GetKernelSubGraphType(const kernel::LiteKernel *kernel, con
}
auto desc = kernel->desc();
if (desc.provider != kernel::kBuiltin) {
return GetCustomKernelSubGraphType(kernel);
}
if (desc.arch == kernel::KERNEL_ARCH::kGPU) {
if (desc.data_type == kNumberTypeFloat16) {
return kernel::kGpuFp16SubGraph;
@ -1228,6 +1212,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;
}