aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/GLES_COMPUTE/GCKernelLibrary.h
diff options
context:
space:
mode:
authorAnthony Barbier <anthony.barbier@arm.com>2017-10-26 15:23:08 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:35:24 +0000
commit7068f9900d136312318ff430aef588b14e0c87ad (patch)
treeb57ca81231860f1d8755e6f18e5be7c959fb60c6 /arm_compute/core/GLES_COMPUTE/GCKernelLibrary.h
parentd60737592736715dcfd0520535c48190d4ac77d2 (diff)
downloadComputeLibrary-7068f9900d136312318ff430aef588b14e0c87ad.tar.gz
COMPMID-631: Merge branches/gles_compute branch
Last commit: commit b25c5f68042b0c81bf611d59a1bb8535e1c42497 Author: Xinghang Zhou <xinghang.zhou@arm.com> Date: Wed Oct 25 18:48:10 2017 +0800 Synced validation's tolerances of GCSoftmax from cl side Change-Id: Ibe72054205c1c8721845d679a31af7ed0a7c5cf6 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/93283 Reviewed-by: Anthony Barbier <anthony.barbier@arm.com> Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'arm_compute/core/GLES_COMPUTE/GCKernelLibrary.h')
-rw-r--r--arm_compute/core/GLES_COMPUTE/GCKernelLibrary.h306
1 files changed, 306 insertions, 0 deletions
diff --git a/arm_compute/core/GLES_COMPUTE/GCKernelLibrary.h b/arm_compute/core/GLES_COMPUTE/GCKernelLibrary.h
new file mode 100644
index 0000000000..e601b529ed
--- /dev/null
+++ b/arm_compute/core/GLES_COMPUTE/GCKernelLibrary.h
@@ -0,0 +1,306 @@
+/*
+ * Copyright (c) 2017 ARM Limited.
+ *
+ * SPDX-License-Identifier: MIT
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to
+ * deal in the Software without restriction, including without limitation the
+ * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
+ * sell copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+#ifndef __ARM_COMPUTE_GCKERNELLIBRARY_H__
+#define __ARM_COMPUTE_GCKERNELLIBRARY_H__
+
+#include "arm_compute/core/GLES_COMPUTE/OpenGLES.h"
+#include "arm_compute/core/Utils.h"
+
+#include <map>
+#include <set>
+#include <string>
+#include <utility>
+#include <vector>
+
+namespace arm_compute
+{
+/** GCProgram class */
+class GCProgram
+{
+public:
+ /** Default constructor. */
+ GCProgram();
+ /** Construct program from source file.
+ *
+ * @param[in] name Program name.
+ * @param[in] source Program source.
+ */
+ GCProgram(std::string name, std::string source);
+ /** Default Copy Constructor. */
+ GCProgram(const GCProgram &) = default;
+ /** Default Move Constructor. */
+ GCProgram(GCProgram &&) = default;
+ /** Default copy assignment operator. */
+ GCProgram &operator=(const GCProgram &) = default;
+ /** Default move assignment operator. */
+ GCProgram &operator=(GCProgram &&) = default;
+ /** Returns program name.
+ *
+ * @return Program's name.
+ */
+ std::string name() const
+ {
+ return _name;
+ }
+ /** Link program.
+ *
+ * @param[in] shader Shader used to link program.
+ *
+ * @return linked program id .
+ */
+ GLuint link_program(GLuint shader);
+ /** Compile shader.
+ *
+ * @param[in] build_options Shader build options.
+ *
+ * @return GLES shader object.
+ */
+ GLuint compile_shader(const std::string &build_options);
+
+private:
+ std::string _name; /**< Program name. */
+ std::string _source; /**< Source code for the program. */
+};
+
+/** GCKernel class */
+class GCKernel
+{
+public:
+ /** Default Constructor. */
+ GCKernel();
+ /** Default Copy Constructor. */
+ GCKernel(const GCKernel &) = default;
+ /** Default Move Constructor. */
+ GCKernel(GCKernel &&) = default;
+ /** Default copy assignment operator. */
+ GCKernel &operator=(const GCKernel &) = default;
+ /** Default move assignment operator. */
+ GCKernel &operator=(GCKernel &&) = default;
+ /** Constructor.
+ *
+ * @param[in] name Kernel name.
+ * @param[in] program Built program.
+ */
+ GCKernel(std::string name, GLuint program);
+ /** Returns kernel name.
+ *
+ * @return Kernel's name.
+ */
+ std::string name() const
+ {
+ return _name;
+ }
+ /** Get program id.
+ *
+ * @return program id.
+ */
+ GLuint get_program() const
+ {
+ return _program;
+ }
+ /** Use current program.
+ *
+ * @return program id.
+ */
+ void use();
+ /** Unuse current program.
+ *
+ * @return program id.
+ */
+ void unuse();
+ /** Set value at uniform idx.
+ *
+ * @param[in] idx Index in vector.
+ * @param[in] value Set value.
+ */
+ template <class T>
+ void set_params(unsigned int idx, T value)
+ {
+ if(idx >= _params.size())
+ {
+ _params.resize(idx + 1, 0);
+ }
+
+ unsigned int *p = reinterpret_cast<unsigned int *>(&value);
+ _params[idx] = *p;
+ }
+ /** Clear params.
+ *
+ */
+ void clear_params()
+ {
+ _params.clear();
+ }
+ /** Set shader params binding point.
+ *
+ * @param[in] binding Shader params binding point.
+ */
+ void set_shader_params_binding_point(unsigned int binding)
+ {
+ _shader_params_binding_point = binding;
+ }
+ /** Update shader params.
+ *
+ */
+ void update_shader_params();
+ /** Clean up program and ubo.
+ *
+ */
+ void cleanup();
+
+private:
+ std::string _name; /**< Kernel name */
+ GLuint _program; /**< Linked program id */
+ std::vector<unsigned int> _params; /**< Store all the values of the shader parameters */
+ GLuint _shader_params; /**< Uniform buffer object name for shader parameters */
+ GLuint _shader_params_binding_point; /**< The binding point of the uniform block for shader parameters */
+ GLuint _shader_params_index; /**< The index of the uniform block */
+ GLint _shader_params_size; /**< The uniform block data size in the shader */
+ static constexpr const char *_shader_params_name = "shader_params"; /**< The uniform block name in the shader */
+};
+
+/** GCKernelLibrary class */
+class GCKernelLibrary
+{
+ using StringSet = std::set<std::string>;
+
+private:
+ /** Default Constructor. */
+ GCKernelLibrary();
+
+public:
+ /** Prevent instances of this class from being copied. */
+ GCKernelLibrary(const GCKernelLibrary &) = delete;
+ /** Prevent instances of this class from being copied. */
+ const GCKernelLibrary &operator=(const GCKernelLibrary &) = delete;
+ /** Default Destructor. */
+ ~GCKernelLibrary();
+
+ static GCKernelLibrary &get();
+ /** Initialises the kernel library.
+ *
+ * @param[in] shader_path (Optional) Path of the directory from which shader sources are loaded.
+ * @param[in] dpy (Optional) EGLdisplay set by external application.
+ * @param[in] ctx (Optional) EGLContext set by external application.
+ */
+ void init(std::string shader_path = "./", EGLDisplay dpy = EGL_NO_DISPLAY, EGLContext ctx = EGL_NO_CONTEXT)
+ {
+ //TODO: deal with old display and context.
+ _shader_path = std::move(shader_path);
+
+ _display = dpy;
+ _context = ctx;
+
+ if(_display == EGL_NO_DISPLAY || _context == EGL_NO_CONTEXT)
+ {
+ setup_context();
+
+ _own_context = true;
+ }
+
+ eglMakeCurrent(_display, EGL_NO_SURFACE, EGL_NO_SURFACE, _context);
+ setup_dummy_fbo();
+ }
+
+ /** Sets the path that the shaders reside in.
+ *
+ * @param[in] shader_path Path of the shader.
+ */
+ void set_shader_path(const std::string &shader_path)
+ {
+ _shader_path = shader_path;
+ };
+ /** Sets display and context to create kernel.
+ *
+ * @param[in] dpy EGLdisplay set by external application.
+ * @param[in] ctx EGLContext set by external application.
+ */
+ void set_context(EGLDisplay dpy, EGLContext ctx)
+ {
+ //TODO: deal with old display and context.
+ _display = dpy;
+ _context = ctx;
+
+ eglMakeCurrent(dpy, EGL_NO_SURFACE, EGL_NO_SURFACE, ctx);
+ setup_dummy_fbo();
+ };
+ /** Creates a kernel from the kernel library.
+ *
+ * @param[in] shader_name Shader name.
+ * @param[in] build_options_set Shader build options as a set.
+ *
+ * @return The created kernel.
+ */
+ GCKernel create_kernel(const std::string &shader_name, const StringSet &build_options_set = {}) const;
+ /** Serializes and saves programs to a binary.
+ *
+ */
+ void save_binary();
+ /** Load serialized binary with all the programs.
+ *
+ */
+ void load_binary();
+ /** Setup a dummy fbo to workaround an issue on Galaxy S8.
+ *
+ */
+ void setup_dummy_fbo();
+
+private:
+ /** Preprocess GLES shader
+ *
+ * @param[in] shader_source Source code of the shader to preprocess.
+ *
+ * @return Preprocessed GLES shader object.
+ */
+ const std::string preprocess_shader(const std::string &shader_source) const;
+ /** Load program and its dependencies.
+ *
+ * @param[in] program_name Name of the program to load.
+ */
+ const GCProgram &load_program(const std::string &program_name) const;
+ /** Concatenates contents of a set into a single string.
+ *
+ * @param[in] s Input set to concatenate.
+ *
+ * @return Concatenated string.
+ */
+ std::string stringify_set(const StringSet &s) const;
+ /** Set up EGL context.
+ */
+ void setup_context();
+
+ EGLDisplay _display; /**< Underlying EGL Display. */
+ EGLContext _context; /**< Underlying EGL Context. */
+ GLuint _frame_buffer; /**< Dummy fbo */
+ GLuint _tex_rt; /**< Dummy texture for render target */
+ bool _own_context; /**< Self created context or not. */
+ std::string _shader_path; /**< Path to the shaders folder. */
+ mutable std::map<std::string, const GCProgram> _programs_map; /**< Map with all already loaded program data. */
+ mutable std::map<std::string, const GCKernel> _built_programs_map; /**< Map with all already built program data. */
+ static const std::map<std::string, std::string> _shader_program_map; /**< Map that associates kernel names with programs. */
+ static const std::map<std::string, std::string> _program_source_map; /**< Contains sources for all programs.
+ Used for compile-time shader inclusion. */
+};
+}
+#endif /* __ARM_COMPUTE_GCKERNELLIBRARY_H__ */