| # Copyright (C) 2020 Arm Limited or its affiliates. All rights reserved. |
| # |
| # SPDX-License-Identifier: Apache-2.0 |
| # |
| # Licensed under the Apache License, Version 2.0 (the License); you may |
| # not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an AS IS BASIS, WITHOUT |
| # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # Description: |
| # Packaging for the Vela compiler |
| from os import path |
| |
| from setuptools import Extension |
| from setuptools import find_namespace_packages |
| from setuptools import setup |
| |
| # Read the contents of README.md file |
| this_directory = path.abspath(path.dirname(__file__)) |
| with open(path.join(this_directory, "PYPI.md"), encoding="utf-8") as f: |
| long_description = f.read() |
| |
| mlw_module = Extension( |
| "ethosu.mlw_codec", |
| ["ethosu/mlw_codec/mlw_encode.c", "ethosu/mlw_codec/mlw_decode.c", "ethosu/mlw_codec/mlw_codecmodule.c"], |
| ) |
| |
| setup( |
| name="ethos-u-vela", |
| use_scm_version=True, |
| description="Optimise TensorFlow Lite models for Ethos-U55 NPU.", |
| long_description=long_description, |
| long_description_content_type="text/markdown", |
| url="https://git.mlplatform.org/ml/ethos-u/ethos-u-vela.git/", |
| author="Arm Ltd.", |
| license="Apache License 2.0", |
| classifiers=[ |
| "Development Status :: 5 - Production/Stable", |
| "Intended Audience :: Developers", |
| "License :: OSI Approved :: Apache Software License", |
| "Operating System :: POSIX :: Linux", |
| "Programming Language :: C", |
| "Programming Language :: Python :: 3", |
| "Programming Language :: Python :: 3.6", |
| "Programming Language :: Python :: 3.7", |
| "Programming Language :: Python :: 3.8", |
| "Topic :: Scientific/Engineering :: Artificial Intelligence", |
| "Topic :: Software Development :: Compilers", |
| ], |
| 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"], |
| entry_points={"console_scripts": ["vela = ethosu.vela.vela:main"]}, |
| ext_modules=[mlw_module], |
| setup_requires=["setuptools_scm"], |
| ) |