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.cpp | |
| 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.cpp')
| -rw-r--r-- | src/backends/cl/ClContextControl.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/cl/ClContextControl.cpp b/src/backends/cl/ClContextControl.cpp index 7ab825f59e..fd2d0f53eb 100644 --- a/src/backends/cl/ClContextControl.cpp +++ b/src/backends/cl/ClContextControl.cpp @@ -28,8 +28,10 @@ namespace armnn { ClContextControl::ClContextControl(arm_compute::CLTuner *tuner, + arm_compute::CLGEMMHeuristicsHandle* heuristicsHandle, bool profilingEnabled) : m_Tuner(tuner) + , m_HeuristicsHandle(heuristicsHandle) , m_ProfilingEnabled(profilingEnabled) { // Ignore m_ProfilingEnabled if unused to avoid compiling problems when ArmCompute is disabled. @@ -156,7 +158,7 @@ void ClContextControl::DoLoadOpenClRuntime(bool updateTunedParameters) // Note the first argument (path to cl source code) will be ignored as they should be embedded in the armcompute. arm_compute::CLKernelLibrary::get().init(".", context, device); - arm_compute::CLScheduler::get().init(context, commandQueue, device, m_Tuner); + arm_compute::CLScheduler::get().init(context, commandQueue, device, m_Tuner, m_HeuristicsHandle); } void ClContextControl::ClearClCache() |