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 | |
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')
-rw-r--r-- | python/pyarmnn/test/test_setup.py | 8 | ||||
-rw-r--r-- | python/pyarmnn/test/test_version.py | 4 |
2 files changed, 6 insertions, 6 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') diff --git a/python/pyarmnn/test/test_version.py b/python/pyarmnn/test/test_version.py index 83606ab15b..f68adff0c7 100644 --- a/python/pyarmnn/test/test_version.py +++ b/python/pyarmnn/test/test_version.py @@ -18,7 +18,7 @@ def test_dev_version(): importlib.reload(v) - assert "30.0.0.dev1" == v.__version__ + assert "31.0.0.dev1" == v.__version__ del os.environ["PYARMNN_DEV_VER"] del v @@ -30,7 +30,7 @@ def test_arm_version_not_affected(): importlib.reload(v) - assert "30.0.0" == v.__arm_ml_version__ + assert "31.0.0" == v.__arm_ml_version__ del os.environ["PYARMNN_DEV_VER"] del v |