commit 4243211cb16d1960b2e1423ebdb83d619cc74b37
author Matthew Sloyan <matthew.sloyan@arm.com> Fri Jan 08 10:30:51 2021 +0000
committer Kevin May <kevin.may@arm.com> Tue Jan 12 10:38:11 2021 +0000
tree 84288d3f689bda2f4ca9ebb0828fe00a7829e81b
parent 80fbcd5f4d7b362360963af1df0121aa6b561576

IVGCVSW-5484 Add CacheLoadedNetwork options to ExecuteNetwork

 * Enable ability to save/load ClContext in ExecuteNetwork.

Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I58c61a53f6713853eb06520cc372ed47baf7f8c4

tests/InferenceModel.hpp

diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp
index 1b87a10..7daae26 100644
--- a/tests/InferenceModel.hpp
+++ b/tests/InferenceModel.hpp
@@ -97,6 +97,8 @@
     bool                            m_ParseUnsupported;
     bool                            m_InferOutputShape;
     bool                            m_EnableFastMath;
+    bool                            m_SaveCachedNetwork;
+    std::string                     m_CachedNetworkFilePath;
 
     Params()
         : m_ComputeDevices{}
@@ -109,6 +111,8 @@
         , m_ParseUnsupported(false)
         , m_InferOutputShape(false)
         , m_EnableFastMath(false)
+        , m_SaveCachedNetwork(false)
+        , m_CachedNetworkFilePath("")
     {}
 };
 
@@ -426,7 +430,9 @@
 
             armnn::BackendOptions gpuAcc("GpuAcc",
             {
-                { "FastMathEnabled", params.m_EnableFastMath }
+                { "FastMathEnabled", params.m_EnableFastMath },
+                { "SaveCachedNetwork", params.m_SaveCachedNetwork },
+                { "CachedNetworkFilePath", params.m_CachedNetworkFilePath }
             });
             armnn::BackendOptions cpuAcc("CpuAcc",
             {