1 files changed, 7 insertions, 5 deletions
diff --git a/src/armnn/Runtime.hpp b/src/armnn/Runtime.hpp
index 694e1e5f9a..e4d4d4ddb9 100644
--- a/src/armnn/Runtime.hpp
+++ b/src/armnn/Runtime.hpp
@@ -10,7 +10,8 @@
 #include <armnn/IRuntime.hpp>
 #include <armnn/Tensor.hpp>
 #include <armnn/BackendId.hpp>
-#include <backendsCommon/IBackendContext.hpp>
+
+#include <cl/ClContextControl.hpp>
 
 #include <mutex>
 #include <unordered_map>
@@ -86,13 +87,14 @@ private:
     }
 
     mutable std::mutex m_Mutex;
+
     std::unordered_map<NetworkId, std::unique_ptr<LoadedNetwork>> m_LoadedNetworks;
-    CreationOptions m_Options;
+
+    ClContextControl m_ClContextControl;
+
     int m_NetworkIdCounter;
-    DeviceSpec m_DeviceSpec;
 
-    using BackendContextMap = std::unordered_map<BackendId, IBackendContextUniquePtr>;
-    BackendContextMap m_BackendContexts;
+    DeviceSpec m_DeviceSpec;
 };
 
 }