diff options
author | Ruomei Yan <ruomei.yan@arm.com> | 2022-08-05 12:01:52 +0100 |
---|---|---|
committer | Ruomei Yan <ruomei.yan@arm.com> | 2022-08-30 17:19:40 +0100 |
commit | 088303393df8bf49e4ea2958e88ff05aa50dd1ec (patch) | |
tree | c8c3bb3872d79c05738033fb98d94ecdb41a1771 | |
parent | 33383a98a2e67c75619e514ffefac91bdaf92abf (diff) | |
download | mlia-088303393df8bf49e4ea2958e88ff05aa50dd1ec.tar.gz |
MLIA-584 Enable testing for aarch64 package
Add conditional dependencies in setup.cfg so that the mlia package can be built in aarch64
Change-Id: I4d878255757818449ba0bda209233c6f9ba9f7d0
-rw-r--r-- | setup.cfg | 11 |
1 files changed, 9 insertions, 2 deletions
@@ -28,10 +28,17 @@ python_requires = >=3.8 package_dir = = src packages = find: +# On aarch64 machine, tensorflow does not pin the numpy version but +# vela does. The pre-built aarch64 tensorflow 2.8.2 uses a numpy whose +# version is conflicting with ethos-u-vela version 3.4.0. Therefore, +# we downgraded both tensorflow and vela versions to make sure there is +# no numpy version conflict. install_requires = - tensorflow~=2.8.2 + tensorflow-aarch64~=2.7.3; platform_machine=="aarch64" + tensorflow~=2.8.2; platform_machine=="x86_64" tensorflow-model-optimization~=0.7.2 - ethos-u-vela~=3.4.0 + ethos-u-vela~=3.3.0; platform_machine=="aarch64" + ethos-u-vela~=3.4.0; platform_machine=="x86_64" requests rich sh |