diff options
| author | Finn Williams <Finn.Williams@arm.com> | 2021-02-11 16:16:42 +0000 |
|---|---|---|
| committer | KeithARM <keith.davis@arm.com> | 2021-02-15 15:04:31 +0000 |
| commit | 406463269f55a5baefb941b51e10f423f6d3250a (patch) | |
| tree | 2299b75c47cae44d9c8be4bf670f30c9d65a01a8 /src/backends/cl/ClContextControl.hpp | |
| parent | b55d366a93d8d3699ce5e906134c695c8040c27e (diff) | |
| download | armnn-406463269f55a5baefb941b51e10f423f6d3250a.tar.gz | |
IVGCVSW-5686 Add GpuAcc MLGO tuning file configuration argument
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I3f320499c379162f9d1b00cc8816bd144cd7eee4
Diffstat (limited to 'src/backends/cl/ClContextControl.hpp')
| -rw-r--r-- | src/backends/cl/ClContextControl.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/backends/cl/ClContextControl.hpp b/src/backends/cl/ClContextControl.hpp index 2ed43bc60b..4a640cdf22 100644 --- a/src/backends/cl/ClContextControl.hpp +++ b/src/backends/cl/ClContextControl.hpp @@ -7,6 +7,7 @@ #include "armnn/IRuntime.hpp" #include <arm_compute/runtime/CL/CLTuner.h> +#include <arm_compute/runtime/CL/CLGEMMHeuristicsHandle.h> namespace armnn { @@ -17,6 +18,7 @@ class ClContextControl public: ClContextControl(arm_compute::CLTuner* = nullptr, + arm_compute::CLGEMMHeuristicsHandle* = nullptr, bool profilingEnabled = false); virtual ~ClContextControl(); @@ -35,6 +37,7 @@ private: void DoLoadOpenClRuntime(bool updateTunedParameters); arm_compute::CLTuner* m_Tuner; + arm_compute::CLGEMMHeuristicsHandle* m_HeuristicsHandle; bool m_ProfilingEnabled; }; @@ -51,6 +54,7 @@ public: TuningLevel m_TuningLevel; arm_compute::CLTuner m_Tuner; + arm_compute::CLGEMMHeuristicsHandle m_HeuristicsHandle; }; } // namespace armnn |