diff options
Diffstat (limited to 'arm_compute/core/CL')
-rw-r--r-- | arm_compute/core/CL/CLTypes.h | 13 | ||||
-rw-r--r-- | arm_compute/core/CL/ICLKernel.h | 9 | ||||
-rw-r--r-- | arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h | 4 |
3 files changed, 23 insertions, 3 deletions
diff --git a/arm_compute/core/CL/CLTypes.h b/arm_compute/core/CL/CLTypes.h index a9d5fdd063..14c31fa396 100644 --- a/arm_compute/core/CL/CLTypes.h +++ b/arm_compute/core/CL/CLTypes.h @@ -60,5 +60,16 @@ enum class CLVersion CL20, /* the OpenCL 2.0 and above */ UNKNOWN /* unkown version */ }; -} + +/** OpenCL device options */ +struct CLDeviceOptions +{ + std::string name; /**< Device name */ + std::string extensions; /**< List of supported extensions */ + std::string ddk_version; /**< DDK version */ + GPUTarget gpu_target; /**< GPU target architecture/instance */ + size_t num_cores; /**< Number of cores */ + size_t cache_size; /**< Cache size */ +}; +} // namespace arm_compute #endif /* __ARM_COMPUTE_CL_TYPES_H__ */ diff --git a/arm_compute/core/CL/ICLKernel.h b/arm_compute/core/CL/ICLKernel.h index c7d0c2156b..e6700cd61e 100644 --- a/arm_compute/core/CL/ICLKernel.h +++ b/arm_compute/core/CL/ICLKernel.h @@ -198,6 +198,15 @@ public: _lws_hint = lws_hint; } + /** Return the Local-Workgroup-Size hint + * + * @return Current lws hint + */ + cl::NDRange lws_hint() const + { + return _lws_hint; + } + /** Get the configuration ID * * @note The configuration ID can be used by the caller to distinguish different calls of the same OpenCL kernel diff --git a/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h b/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h index d47b7da213..eb1bf58b1b 100644 --- a/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h +++ b/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -88,7 +88,7 @@ public: void run(const Window &window, cl::CommandQueue &queue) override; BorderSize border_size() const override; -private: +public: const ICLTensor *_input; const ICLTensor *_biases; const ICLTensor *_weights; |