MLECO-712 Enable creation of runtime with external profiling support
Change-Id: Ie0af439817eb4ed70e6112b2eb6f4a2c787a6bd0
Signed-off-by: Éanna Ó Catháin <eanna.ocathain@arm.com>
diff --git a/python/pyarmnn/test/test_runtime.py b/python/pyarmnn/test/test_runtime.py
index 2943be8..ff0ad40 100644
--- a/python/pyarmnn/test/test_runtime.py
+++ b/python/pyarmnn/test/test_runtime.py
@@ -14,6 +14,7 @@
network = parser.CreateNetworkFromBinaryFile(os.path.join(shared_data_folder, 'mock_model.tflite'))
preferred_backends = [ann.BackendId('CpuRef')]
options = ann.CreationOptions()
+
runtime = ann.IRuntime(options)
graphs_count = parser.GetSubgraphCount()
@@ -95,7 +96,6 @@
assert not opt_network.thisown
-
def test_load_network(random_runtime):
preferred_backends = random_runtime[0]
network = random_runtime[1]
@@ -108,6 +108,40 @@
assert "" == messages
assert net_id == 0
+def test_create_runtime_with_external_profiling_enabled():
+
+ options = ann.CreationOptions()
+
+ options.m_ProfilingOptions.m_FileOnly = True
+ options.m_ProfilingOptions.m_EnableProfiling = True
+ options.m_ProfilingOptions.m_OutgoingCaptureFile = "/tmp/outgoing.txt"
+ options.m_ProfilingOptions.m_IncomingCaptureFile = "/tmp/incoming.txt"
+ options.m_ProfilingOptions.m_TimelineEnabled = True
+ options.m_ProfilingOptions.m_CapturePeriod = 1000
+ options.m_ProfilingOptions.m_FileFormat = "JSON"
+
+ runtime = ann.IRuntime(options)
+
+ assert runtime is not None
+
+def test_create_runtime_with_external_profiling_enabled_invalid_options():
+
+ options = ann.CreationOptions()
+
+ options.m_ProfilingOptions.m_FileOnly = True
+ options.m_ProfilingOptions.m_EnableProfiling = False
+ options.m_ProfilingOptions.m_OutgoingCaptureFile = "/tmp/outgoing.txt"
+ options.m_ProfilingOptions.m_IncomingCaptureFile = "/tmp/incoming.txt"
+ options.m_ProfilingOptions.m_TimelineEnabled = True
+ options.m_ProfilingOptions.m_CapturePeriod = 1000
+ options.m_ProfilingOptions.m_FileFormat = "JSON"
+
+ with pytest.raises(RuntimeError) as err:
+ runtime = ann.IRuntime(options)
+
+ expected_error_message = "It is not possible to enable timeline reporting without profiling being enabled"
+ assert expected_error_message in str(err.value)
+
def test_load_network_properties_provided(random_runtime):
preferred_backends = random_runtime[0]