aboutsummaryrefslogtreecommitdiff
path: root/setup.py
diff options
context:
space:
mode:
Diffstat (limited to 'setup.py')
-rw-r--r--setup.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/setup.py b/setup.py
index 805062b0..bb2e8a09 100644
--- a/setup.py
+++ b/setup.py
@@ -76,7 +76,12 @@ setup(
keywords=["ethos-u", "vela compiler", "tflite", "npu"],
packages=find_namespace_packages(include=["ethosu.*"]),
python_requires="~=3.6", # We support only 3.6+
- install_requires=["flatbuffers==1.11.0", "numpy>=1.16.6", "lxml>=4.5.1"],
+ install_requires=[
+ "flatbuffers==1.11.0",
+ "numpy>=1.16.6",
+ "numpy>=1.16.6,<1.19.4 ; platform_system=='Windows'",
+ "lxml>=4.5.1",
+ ],
entry_points={"console_scripts": ["vela = ethosu.vela.vela:main"]},
ext_modules=[mlw_module, tensor_allocator_module],
setup_requires=["setuptools_scm"],