IVGCVSW-5317 'Add enable_fast_math Option to ExecuteNetwork'

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I4eb3e27837aea926593d49f9ccea07bab8388d5b

1 files changed, 13 insertions, 0 deletions

diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp
index 527cc64d85..0a458c8b64 100644
--- a/tests/InferenceModel.hpp
+++ b/tests/InferenceModel.hpp
@@ -97,6 +97,7 @@ struct Params
     bool                            m_PrintIntermediateLayers;
     bool                            m_ParseUnsupported;
     bool                            m_InferOutputShape;
+    bool                            m_EnableFastMath;
 
     Params()
         : m_ComputeDevices{}
@@ -108,6 +109,7 @@ struct Params
         , m_PrintIntermediateLayers(false)
         , m_ParseUnsupported(false)
         , m_InferOutputShape(false)
+        , m_EnableFastMath(false)
     {}
 };
 
@@ -418,6 +420,17 @@ public:
             options.m_ReduceFp32ToBf16 = params.m_EnableBf16TurboMode;
             options.m_Debug = params.m_PrintIntermediateLayers;
 
+            armnn::BackendOptions gpuAcc("GpuAcc",
+            {
+                { "FastMathEnabled", params.m_EnableFastMath }
+            });
+            armnn::BackendOptions cpuAcc("CpuAcc",
+            {
+                { "FastMathEnabled", params.m_EnableFastMath }
+            });
+            options.m_ModelOptions.push_back(gpuAcc);
+            options.m_ModelOptions.push_back(cpuAcc);
+
             const auto optimization_start_time = armnn::GetTimeNow();
             optNet = armnn::Optimize(*network, params.m_ComputeDevices, m_Runtime->GetDeviceSpec(), options);