-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
39 lines (33 loc) · 1.03 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
#!/usr/bin/env python
from setuptools import setup, find_packages
from setuptools.extension import Extension
from Cython.Build import cythonize
import numpy as np
import os
MAJOR = 0
MINOR = 1
MICRO = 0
ISRELEASED = False
VERSION = '%d.%d.%d' % (MAJOR, MINOR, MICRO)
reqs = []
if os.path.isfile('requirements.txt'):
with open('requirements.txt') as f:
required = f.read().splitlines()
reqs.extend(required)
params = {}
extensions = [Extension("loops", ["c/loops.pyx"],
#extra_compile_args=['-ffast-math'],
#extra_link_args=['-O3'],
include_dirs=[np.get_include()])]
setup(name='fast-fusion',
version=VERSION,
description='Fast methods for collective matrix tri-factorization.',
url='http://gitlab.fri.uni-lj.si/acopar/fast-fusion',
author='Andrej Copar',
license='LGPL',
packages=find_packages(),
ext_modules=cythonize(extensions),
zip_safe=False,
install_requires=reqs,
**params
)