aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/CL
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/core/CL')
-rw-r--r--arm_compute/core/CL/CLKernelLibrary.h6
1 files changed, 6 insertions, 0 deletions
diff --git a/arm_compute/core/CL/CLKernelLibrary.h b/arm_compute/core/CL/CLKernelLibrary.h
index f25915f053..18b6bb4434 100644
--- a/arm_compute/core/CL/CLKernelLibrary.h
+++ b/arm_compute/core/CL/CLKernelLibrary.h
@@ -277,6 +277,12 @@ public:
return _context;
}
+ /** Gets the CL device for which the programs are created. */
+ cl::Device &get_device()
+ {
+ return _device;
+ }
+
/** Sets the CL device for which the programs are created.
*
* @param[in] device A CL device.