forked from richarao/hls4ml
-
Notifications
You must be signed in to change notification settings - Fork 3
/
setup.py
50 lines (47 loc) · 1.77 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
from setuptools import setup
from setuptools import find_packages
long_description = '''
A package for machine learning inference in FPGAs. We create firmware
implementations of machine learning algorithms using high level synthesis
language (HLS). We translate traditional open-source machine learning package
models into HLS that can be configured for your use-case!
For more information visit the webpage:
https://hls-fpga-machine-learning.github.io/hls4ml/
'''
setup(name='hls4ml',
version='0.2.0',
description='Machine learning in FPGAs using HLS',
long_description=long_description,
author='HLS4ML Team',
author_email='hls4ml.help@gmail.com',
url='https://github.com/hls-fpga-machine-learning/hls4ml',
license='Apache 2.0',
install_requires=['numpy',
'six',
'pyyaml',
'h5py',
'onnx>=1.4.0'],
extras_require={
'profiling': [
'pandas',
'seaborn',
'matplotlib',
'tensorflow'
]
},
scripts=['scripts/hls4ml'],
include_package_data=True,
classifiers=[
'Development Status :: 3 - Alpha',
'Intended Audience :: Developers',
'Intended Audience :: Science/Research',
'License :: OSI Approved :: Apache Software License',
'Programming Language :: C++',
'Programming Language :: Python :: 2',
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.6',
'Topic :: Software Development :: Libraries',
'Topic :: Software Development :: Libraries :: Python Modules'
],
packages=find_packages())