From 5383767a7a759c867235ab66bd71f88281e3bd06 Mon Sep 17 00:00:00 2001 From: Cathal Corbett Date: Thu, 1 Sep 2022 11:34:37 +0100 Subject: Optimize the calling of IsLayerSupported(). * Done as part of 22.11/23.02 innovation days. * IsLayerSupported() is called in model prepare (delegate, android-nn-driver and shim/support_library) and again in ArmNN once model otimization is performed. * From calling IsLayerSupported() the first time, we should know that the layers are supported and what backend they are supported on. * Solution is to set the BackendId of the IConnectableLayer when IsLayerSupported() is called the first time, * In the Optimize() function we then check if the backend is set. If so, we do not call IsLayerSupported() again. * In the case a layer that is supported gets optimized, then the BackendId of that layer get set to "Unknown" for the new optimized layer and IsLayerSupported() will get called on the newly optimized layer. * Includes bug fix IVGCVSW-7213 for Android Mean FP16 CpuAcc tests. Also related to bug IVGCVSW-7211. Signed-off-by: Cathal Corbett Change-Id: I7a7820d0cdb079ffb5a3a2e0c44e252f652df53b --- python/pyarmnn/test/test_setup.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'python/pyarmnn/test/test_setup.py') diff --git a/python/pyarmnn/test/test_setup.py b/python/pyarmnn/test/test_setup.py index ada96ccad4..8275a537b4 100644 --- a/python/pyarmnn/test/test_setup.py +++ b/python/pyarmnn/test/test_setup.py @@ -87,15 +87,15 @@ def test_gcc_serch_path(): def test_armnn_version(): - check_armnn_version('31.0.0', '31.0.0') + check_armnn_version('32.0.0', '32.0.0') def test_incorrect_armnn_version(): with pytest.raises(AssertionError) as err: - check_armnn_version('31.0.0', '31.1.0') + check_armnn_version('32.0.0', '32.1.0') - assert 'Expected ArmNN version is 31.1.0 but installed ArmNN version is 31.0.0' in str(err.value) + assert 'Expected ArmNN version is 32.1.0 but installed ArmNN version is 32.0.0' in str(err.value) def test_armnn_version_patch_does_not_matter(): - check_armnn_version('31.0.0', '31.0.1') + check_armnn_version('32.0.0', '32.0.1') -- cgit v1.2.1