aboutsummaryrefslogtreecommitdiff
path: root/python
diff options
context:
space:
mode:
authorNina Drozd <nina.drozd@arm.com>2020-12-01 11:56:12 +0000
committerJim Flynn <jim.flynn@arm.com>2020-12-08 11:47:32 +0000
commit6249d7e5f74323d2322fd69409db616efe46f8c8 (patch)
treeeb7e781d37264873a65c87eb19bc165d51f4eb16 /python
parent85080fc88aad0a5de13ce5b878e6277c4c03032c (diff)
downloadarmnn-6249d7e5f74323d2322fd69409db616efe46f8c8.tar.gz
Pyarmnn fix - parsers are optional extensions
Signed-off-by: Nina Drozd <nina.drozd@arm.com> Change-Id: Ic999cb898933fa674f1cd3b4b85ab650154d587d
Diffstat (limited to 'python')
-rwxr-xr-xpython/pyarmnn/setup.py18
1 files changed, 10 insertions, 8 deletions
diff --git a/python/pyarmnn/setup.py b/python/pyarmnn/setup.py
index 94dd63ea57..e1eba4b082 100755
--- a/python/pyarmnn/setup.py
+++ b/python/pyarmnn/setup.py
@@ -186,11 +186,13 @@ class LazyArmnnFinderExtension(Extension):
self._library_dirs = None
self._runtime_library_dirs = None
self._armnn_libs = armnn_libs
- self._optional = optional[0]
- # self.__swig_opts = None
- super().__init__(name, sources, include_dirs, define_macros, undef_macros, library_dirs, libraries,
- runtime_library_dirs, extra_objects, extra_compile_args, extra_link_args, export_symbols,
- language, optional, **kw)
+ self._optional = False if optional is None else optional
+
+ super().__init__(name=name, sources=sources, include_dirs=include_dirs, define_macros=define_macros,
+ undef_macros=undef_macros, library_dirs=library_dirs, libraries=libraries,
+ runtime_library_dirs=runtime_library_dirs, extra_objects=extra_objects,
+ extra_compile_args=extra_compile_args, extra_link_args=extra_link_args,
+ export_symbols=export_symbols, language=language, optional=optional, **kw)
@property
def include_dirs(self):
@@ -256,14 +258,14 @@ if __name__ == '__main__':
extra_compile_args=['-std=c++14'],
language='c++',
armnn_libs=['libarmnn.so'],
- optional=[False]
+ optional=False
)
pyarmnn_v_module = LazyArmnnFinderExtension('pyarmnn._generated._pyarmnn_version',
sources=['src/pyarmnn/_generated/armnn_version_wrap.cpp'],
extra_compile_args=['-std=c++14'],
language='c++',
armnn_libs=['libarmnn.so'],
- optional=[False]
+ optional=False
)
extensions_to_build = [pyarmnn_v_module, pyarmnn_module]
@@ -276,7 +278,7 @@ if __name__ == '__main__':
extra_compile_args=['-std=c++14'],
language='c++',
armnn_libs=['libarmnn.so', 'libarmnn{}.so'.format(name)],
- optional=[True]
+ optional=True
)
ext_list.append(pyarmnn_optional_module)