diff options
author | Pablo Tello <pablo.tello@arm.com> | 2019-09-24 11:03:47 +0100 |
---|---|---|
committer | Pablo Marquez <pablo.tello@arm.com> | 2019-10-15 14:05:55 +0000 |
commit | db8485ac24135f17e9882c76196924435abc064f (patch) | |
tree | dfe4ff6a50012ac93c6b1cf3fb29c099a7592522 /src/core/CL/CLKernelLibrary.cpp | |
parent | a046e164b96a8441b2fa14ef578f7db46a0e97da (diff) | |
download | ComputeLibrary-db8485ac24135f17e9882c76196924435abc064f.tar.gz |
COMPMID-2205: CL runtime context.
CL Interfaces implemented.
Concrete classes implemented.
One test (ActivationLayer) ported to the new interface.
Change-Id: I283808bec36ccfc2f13fe048c45cbbee698ce525
Signed-off-by: Pablo Tello <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1998
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/CL/CLKernelLibrary.cpp')
-rw-r--r-- | src/core/CL/CLKernelLibrary.cpp | 65 |
1 files changed, 65 insertions, 0 deletions
diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp index 7b7263fca7..c27f886129 100644 --- a/src/core/CL/CLKernelLibrary.cpp +++ b/src/core/CL/CLKernelLibrary.cpp @@ -1144,6 +1144,49 @@ Kernel CLKernelLibrary::create_kernel(const std::string &kernel_name, const Stri return Kernel(kernel_name, cl_program); } +void CLKernelLibrary::init(std::string kernel_path, cl::Context context, cl::Device device) +{ + _kernel_path = std::move(kernel_path); + _context = std::move(context); + _device = std::move(device); +} + +void CLKernelLibrary::set_kernel_path(const std::string &kernel_path) +{ + _kernel_path = kernel_path; +} + +cl::Context &CLKernelLibrary::context() +{ + return _context; +} + +cl::Device &CLKernelLibrary::get_device() +{ + return _device; +} + +void CLKernelLibrary::set_device(cl::Device device) +{ + _device = std::move(device); +} + +std::string CLKernelLibrary::get_kernel_path() +{ + return _kernel_path; +} + +void CLKernelLibrary::clear_programs_cache() +{ + _programs_map.clear(); + _built_programs_map.clear(); +} + +const std::map<std::string, cl::Program> &CLKernelLibrary::get_built_programs() const +{ + return _built_programs_map; +} + void CLKernelLibrary::add_built_program(const std::string &built_program_name, const cl::Program &program) { _built_programs_map.emplace(built_program_name, program); @@ -1205,6 +1248,28 @@ const Program &CLKernelLibrary::load_program(const std::string &program_name) co return new_program.first->second; } +void CLKernelLibrary::set_context(cl::Context context) +{ + _context = std::move(context); + if(_context.get() == nullptr) + { + _device = cl::Device(); + } + else + { + const auto cl_devices = _context.getInfo<CL_CONTEXT_DEVICES>(); + + if(cl_devices.empty()) + { + _device = cl::Device(); + } + else + { + _device = cl_devices[0]; + } + } +} + std::string CLKernelLibrary::stringify_set(const StringSet &s) const { std::string concat_set; |