diff options
author | Samuel Yap <samuel.yap@arm.com> | 2022-08-08 14:07:42 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2022-08-30 17:03:33 +0100 |
commit | dc8ed9d75e54e914a970e137900930fa64a0782b (patch) | |
tree | 8bcaedaae81a6afbdbe3c9a4e69e45840f18cdb4 /python/pyarmnn/test/test_setup.py | |
parent | 9c9d5b9d796d243d88bd7a7aebb2e7e6c467e3a4 (diff) | |
download | armnn-dc8ed9d75e54e914a970e137900930fa64a0782b.tar.gz |
IVGCVSW-7105: BatchMatMul Optional Parameter Support
* Added transpose parameters to pre-transpose each input tensor's slices
* Added adjoint parameters to pre-adjoint each input tensor's slices
* Small refactoring (BatchMatMulDescriptor static helpers and BatchMatMulImpl constructor)
* Updated input validation and output shape inference for parameters
* Additional layer unit tests for parameters added
* Versionings incremented
Signed-off-by: Samuel Yap <samuel.yap@arm.com>
Change-Id: Ibe5242a8a5bf604c13de0dc65844fd6c421cc667
Diffstat (limited to 'python/pyarmnn/test/test_setup.py')
-rw-r--r-- | python/pyarmnn/test/test_setup.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/python/pyarmnn/test/test_setup.py b/python/pyarmnn/test/test_setup.py index 27feda2647..ada96ccad4 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('30.0.0', '30.0.0') + check_armnn_version('31.0.0', '31.0.0') def test_incorrect_armnn_version(): with pytest.raises(AssertionError) as err: - check_armnn_version('30.0.0', '30.1.0') + check_armnn_version('31.0.0', '31.1.0') - assert 'Expected ArmNN version is 30.1.0 but installed ArmNN version is 30.0.0' in str(err.value) + assert 'Expected ArmNN version is 31.1.0 but installed ArmNN version is 31.0.0' in str(err.value) def test_armnn_version_patch_does_not_matter(): - check_armnn_version('30.0.0', '30.0.1') + check_armnn_version('31.0.0', '31.0.1') |