diff options
Diffstat (limited to 'arm_compute/core/GLES_COMPUTE')
-rw-r--r-- | arm_compute/core/GLES_COMPUTE/GCHelpers.h | 38 | ||||
-rw-r--r-- | arm_compute/core/GLES_COMPUTE/IGCKernel.h | 20 |
2 files changed, 27 insertions, 31 deletions
diff --git a/arm_compute/core/GLES_COMPUTE/GCHelpers.h b/arm_compute/core/GLES_COMPUTE/GCHelpers.h index 8fbb5bbb2e..1422afe2b5 100644 --- a/arm_compute/core/GLES_COMPUTE/GCHelpers.h +++ b/arm_compute/core/GLES_COMPUTE/GCHelpers.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -25,43 +25,19 @@ #define __ARM_COMPUTE_GCHELPERS_H__ #include "arm_compute/core/GLES_COMPUTE/OpenGLES.h" +#include "arm_compute/core/GPUTarget.h" #include "arm_compute/core/Helpers.h" #include "support/ToolchainSupport.h" -#include <string> - namespace arm_compute { -namespace gles_compute -{ -/** Helper function to create and return a unique_ptr pointed to a GLES kernel object - * It also calls the kernel's configuration. - * - * @param[in] args All the arguments that need pass to kernel's configuration. - * - * @return A unique pointer pointed to a GLES kernel object - */ -template <typename Kernel, typename... T> -std::unique_ptr<Kernel> create_configure_kernel(T &&... args) -{ - std::unique_ptr<Kernel> k = arm_compute::support::cpp14::make_unique<Kernel>(); - k->configure(std::forward<T>(args)...); - return k; -} +/** Max vector width of an GLES vector */ +static constexpr unsigned int max_gc_vector_width = 16; -/** Helper function to create and return a unique_ptr pointed to a GLES kernel object +/** Helper function to get the GPU target from GLES using GL_RENDERER enum * - * @return A unique pointer pointed to a GLES kernel object + * @return the GPU target */ -template <typename Kernel> -std::unique_ptr<Kernel> create_kernel() -{ - std::unique_ptr<Kernel> k = arm_compute::support::cpp14::make_unique<Kernel>(); - return k; -} - -/** Max vector width of an GLES vector */ -static constexpr unsigned int max_gc_vector_width = 16; -} // namespace gles_compute +GPUTarget get_target_from_device(); } // namespace arm_compute #endif /* __ARM_COMPUTE_GCHELPERS_H__ */ diff --git a/arm_compute/core/GLES_COMPUTE/IGCKernel.h b/arm_compute/core/GLES_COMPUTE/IGCKernel.h index 3b987330da..79d579365c 100644 --- a/arm_compute/core/GLES_COMPUTE/IGCKernel.h +++ b/arm_compute/core/GLES_COMPUTE/IGCKernel.h @@ -26,6 +26,7 @@ #include "arm_compute/core/GLES_COMPUTE/GCKernelLibrary.h" #include "arm_compute/core/GLES_COMPUTE/OpenGLES.h" +#include "arm_compute/core/GPUTarget.h" #include "arm_compute/core/IKernel.h" @@ -105,6 +106,24 @@ public: _lws_hint = lws_hint; } + /** Set the targeted GPU architecture + * + * @param[in] target The targeted GPU architecture + */ + void set_target(GPUTarget target) + { + _target = target; + } + + /** Get the targeted GPU architecture + * + * @return The targeted GPU architecture. + */ + GPUTarget get_target() const + { + return _target; + } + private: /** Add the passed tensor's parameters to the object's kernel's arguments starting from the index idx. * @@ -126,6 +145,7 @@ private: protected: GCKernel _kernel; /**< GLES kernel to run */ gles::NDRange _lws_hint; /**< Local workgroup size hint for the GLES kernel */ + GPUTarget _target; /**< The targeted GPU */ }; /** Add the kernel to the command queue with the given window. |