diff options
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/test/RuntimeTests.cpp | 6 | ||||
-rw-r--r-- | src/armnn/test/TestUtils.cpp | 10 | ||||
-rw-r--r-- | src/armnn/test/TestUtils.hpp | 8 |
3 files changed, 19 insertions, 5 deletions
diff --git a/src/armnn/test/RuntimeTests.cpp b/src/armnn/test/RuntimeTests.cpp index 9ced7e907c..d5bc26a72e 100644 --- a/src/armnn/test/RuntimeTests.cpp +++ b/src/armnn/test/RuntimeTests.cpp @@ -21,6 +21,7 @@ #include <boost/test/unit_test.hpp> #include "RuntimeTests.hpp" +#include "TestUtils.hpp" namespace armnn { @@ -30,11 +31,6 @@ void RuntimeLoadedNetworksReserve(armnn::Runtime* runtime) runtime->m_LoadedNetworks.reserve(1); } -profiling::ProfilingService& GetProfilingService(armnn::Runtime* runtime) -{ - return runtime->m_ProfilingService; -} - } BOOST_AUTO_TEST_SUITE(Runtime) diff --git a/src/armnn/test/TestUtils.cpp b/src/armnn/test/TestUtils.cpp index b0ed17e41a..8ef820b3d5 100644 --- a/src/armnn/test/TestUtils.cpp +++ b/src/armnn/test/TestUtils.cpp @@ -18,3 +18,13 @@ void Connect(armnn::IConnectableLayer* from, armnn::IConnectableLayer* to, const from->GetOutputSlot(fromIndex).Connect(to->GetInputSlot(toIndex)); from->GetOutputSlot(fromIndex).SetTensorInfo(tensorInfo); } + +namespace armnn +{ + +profiling::ProfilingService& GetProfilingService(armnn::Runtime* runtime) +{ + return runtime->m_ProfilingService; +} + +}
\ No newline at end of file diff --git a/src/armnn/test/TestUtils.hpp b/src/armnn/test/TestUtils.hpp index 9b1a3bf18d..9c5f672a5a 100644 --- a/src/armnn/test/TestUtils.hpp +++ b/src/armnn/test/TestUtils.hpp @@ -7,6 +7,7 @@ #include <armnn/INetwork.hpp> #include <Graph.hpp> +#include <Runtime.hpp> void Connect(armnn::IConnectableLayer* from, armnn::IConnectableLayer* to, const armnn::TensorInfo& tensorInfo, unsigned int fromIndex = 0, unsigned int toIndex = 0); @@ -47,3 +48,10 @@ bool CheckRelatedLayers(armnn::Graph& graph, const std::list<std::string>& testR return true; } + +namespace armnn +{ + +profiling::ProfilingService& GetProfilingService(armnn::Runtime* runtime); + +} // namespace armnn
\ No newline at end of file |