From 063f3e3d26f5558c45b332d5cf3ce0fa9154427d Mon Sep 17 00:00:00 2001
From: jianghui58 <jianghui58@huawei.com>
Date: Fri, 26 Nov 2021 11:07:31 +0800
Subject: [PATCH] modify get custom kernel cpu subgraph type

---
 mindspore/lite/src/scheduler.cc | 18 ++----------------
 1 file changed, 2 insertions(+), 16 deletions(-)

diff --git a/mindspore/lite/src/scheduler.cc b/mindspore/lite/src/scheduler.cc
index eed000fa9ba..24e671131cf 100644
--- a/mindspore/lite/src/scheduler.cc
+++ b/mindspore/lite/src/scheduler.cc
@@ -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;
 }