-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
92 lines (86 loc) · 4.6 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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
# Copyright 2019-2021 ETH Zurich and the DaCe authors. All rights reserved.
import glob
import os
import shutil
import subprocess
from setuptools import find_packages, setup
# Find runtime and external library files by obtaining the module path and
# trimming the absolute path of the resulting files.
dace_path = os.path.dirname(os.path.abspath(__file__)) + '/dace/'
runtime_files = [f[len(dace_path):] for f in glob.glob(dace_path + 'runtime/include/**/*', recursive=True)]
library_files = [f[len(dace_path):] for f in glob.glob(dace_path + 'libraries/**/include/**/*', recursive=True)]
cmake_files = [f[len(dace_path):] for f in glob.glob(dace_path + 'codegen/**/*.cmake', recursive=True)]
viewer_files = [
f[len(dace_path):] for f in (glob.glob(dace_path + 'viewer/webclient/dist/*.js', recursive=True) +
glob.glob(dace_path + 'viewer/webclient/external_libs/**/*', recursive=True) +
glob.glob(dace_path + 'viewer/webclient/*.css', recursive=True) +
glob.glob(dace_path + 'viewer/webclient/*.html', recursive=True) +
glob.glob(dace_path + 'viewer/templates/**/*', recursive=True) +
glob.glob(dace_path + 'viewer/**/LICENSE', recursive=True))
]
cub_files = [f[len(dace_path):] for f in glob.glob(dace_path + 'external/cub/cub/**/*', recursive=True)
] + [dace_path + 'external/cub/LICENSE.TXT']
hlslib_files = [f[len(dace_path):] for f in glob.glob(dace_path + 'external/hlslib/cmake/**/*', recursive=True)] + [
f[len(dace_path):] for f in glob.glob(dace_path + 'external/hlslib/include/**/*', recursive=True)
] + [dace_path + 'external/hlslib/LICENSE.md']
rtllib_files = [f[len(dace_path):] for f in glob.glob(dace_path + 'external/rtllib/cmake/**/*', recursive=True)] + [
f[len(dace_path):] for f in glob.glob(dace_path + 'external/rtllib/templates/**/*', recursive=True)
]
# See if CMake is available and if not, install as a dependency
cmake_requires = ['scikit-build', 'cmake']
try:
cmake_path = shutil.which('cmake')
if cmake_path:
# CMake is available, check version
output = subprocess.check_output([cmake_path, '--version']).decode('utf-8')
cmake_version = tuple(int(t) for t in output.splitlines()[0].split(' ')[-1].split('.'))
# If version meets minimum requirements, CMake is not necessary
if cmake_version >= (3, 15):
cmake_requires = []
except (subprocess.CalledProcessError, OSError, IndexError, ValueError):
# Any failure in getting the CMake version counts as "not found"
pass
with open("README.md", "r") as fp:
long_description = fp.read()
with open(os.path.join(dace_path, "version.py"), "r") as fp:
version = fp.read().strip().split(' ')[-1][1:-1]
setup(name='dace',
version=version,
url='https://github.com/spcl/dace',
author='SPCL @ ETH Zurich',
author_email='[email protected]',
description='Data-Centric Parallel Programming Framework',
long_description=long_description,
long_description_content_type='text/markdown',
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: BSD License",
"Operating System :: OS Independent",
],
python_requires='>=3.6, <3.12',
packages=find_packages(exclude=["*.tests", "*.tests.*", "tests.*", "tests"]),
package_data={
'': [
'*.yml', 'codegen/CMakeLists.txt', 'codegen/tools/*.cpp', 'external/moodycamel/*.h',
'external/moodycamel/LICENSE.md', 'codegen/Xilinx_HLS.tcl.in'
] + runtime_files + cub_files + viewer_files + hlslib_files + library_files + rtllib_files + cmake_files
},
include_package_data=True,
install_requires=[
'numpy', 'networkx >= 2.5', 'astunparse', 'sympy<=1.9', 'pyyaml', 'ply', 'websockets', 'requests', 'flask',
'fparser >= 0.1.3', 'aenum >= 3.1', 'dataclasses; python_version < "3.7"', 'dill',
'pyreadline;platform_system=="Windows"', 'typing-compat; python_version < "3.8"'
] + cmake_requires,
extras_require={
'testing': ['coverage', 'pytest-cov', 'scipy', 'absl-py', 'opt_einsum', 'pymlir', 'click'],
'docs': ['jinja2<3.1.0', 'sphinx-autodoc-typehints', 'sphinx-rtd-theme>=0.5.1']
},
entry_points={
'console_scripts': [
'dacelab = dace.cli.dacelab:main',
'sdfv = dace.cli.sdfv:main',
'sdfgcc = dace.cli.sdfgcc:main',
'fcfd = dace.cli.fcdc:main',
'daceprof = dace.cli.daceprof:main',
],
})