From a59c5c1097e9ba29013f4e90b8f4b68475214de7 Mon Sep 17 00:00:00 2001
From: "ysh329@sina.com" <ysh329@sina.com>
Date: Wed, 8 Apr 2020 09:51:59 +0000
Subject: [PATCH] Add cl context. test=develop

---
 lite/backends/opencl/cl_context.cc | 18 +++++++++++++++++-
 lite/backends/opencl/cl_context.h  |  5 +++++
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/lite/backends/opencl/cl_context.cc b/lite/backends/opencl/cl_context.cc
index 153c062003..75a3479541 100644
--- a/lite/backends/opencl/cl_context.cc
+++ b/lite/backends/opencl/cl_context.cc
@@ -45,7 +45,6 @@ cl::Program &CLContext::GetProgram(const std::string &file_name,
   }
 
   auto program = CLRuntime::Global()->CreateProgram(GetContext(), file_name);
-
   VLOG(3) << " --- begin build program -> " << program_key << " --- ";
   CLRuntime::Global()->BuildProgram(program.get(), options);
   VLOG(3) << " --- end build program -> " << program_key << " --- ";
@@ -77,6 +76,23 @@ void CLContext::AddKernel(const std::string &kernel_name,
   kernel_offset_map[kernel_key.str()] = kernels.size() - 1;
 }
 
+std::shared_ptr<cl::Kernel> CLContext::CreateKernel(
+    const std::string &kernel_name,
+    const std::string &file_name,
+    const std::string &options,
+    const std::string &time_stamp) {
+  cl_int status{CL_SUCCESS};
+  VLOG(3) << " --- to get program " << file_name << " --- ";
+  auto program = GetProgram(file_name, options);
+  VLOG(3) << " --- end get program --- ";
+  VLOG(3) << " --- to create kernel: " << kernel_name << " --- ";
+  std::shared_ptr<cl::Kernel> kernel(
+      new cl::Kernel(program, kernel_name.c_str(), &status));
+  CL_CHECK_FATAL(status);
+  VLOG(3) << " --- end create kernel --- ";
+  return kernel;
+}
+
 cl::Kernel &CLContext::GetKernel(const int index) {
   auto &kernels = CLRuntime::Global()->kernels();
   VLOG(3) << " --- kernel count: " << kernels.size() << " --- ";
diff --git a/lite/backends/opencl/cl_context.h b/lite/backends/opencl/cl_context.h
index b12473ccf5..4d609ab3a4 100644
--- a/lite/backends/opencl/cl_context.h
+++ b/lite/backends/opencl/cl_context.h
@@ -39,6 +39,11 @@ class CLContext {
                  const std::string &options = "",
                  const std::string &time_stamp = "");
 
+  std::shared_ptr<cl::Kernel> CreateKernel(const std::string &kernel_name,
+                                           const std::string &file_name,
+                                           const std::string &options = "",
+                                           const std::string &time_stamp = "");
+
   cl::Kernel &GetKernel(const int index);
 
   cl::Kernel &GetKernel(const std::string &name);
-- 
GitLab