aboutsummaryrefslogtreecommitdiff
path: root/src/armnn
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn')
-rw-r--r--src/armnn/test/RuntimeTests.cpp6
-rw-r--r--src/armnn/test/TestUtils.cpp10
-rw-r--r--src/armnn/test/TestUtils.hpp8
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