try:
from setuptools import setup
from setuptools import Extension
except:
from distutils.core import setup
from distutils.extension import Extension
from Cython.Build import cythonize
from Cython.Distutils import build_ext
import numpy
import platform
## https://stackoverflow.com/questions/724664/python-distutils-how-to-get-a-compiler-that-is-going-to-be-used
class build_ext_subclass( build_ext ):
def build_extensions(self):
c = self.compiler.compiler_type
# TODO: add entries for intel's ICC
if c == 'msvc': # visual studio
for e in self.extensions:
e.extra_compile_args = ['/openmp', '/O2']
else: # gcc and clang
for e in self.extensions:
e.extra_compile_args = ['-fopenmp', '-O2', '-march=native', '-std=c99']
e.extra_link_args = ['-fopenmp']
### Comment: -Ofast gives worse speed than -O2 or -O3
build_ext.build_extensions(self)
setup(
name = 'hpfrec',
packages = ['hpfrec'],
install_requires=[
'pandas>=0.24',
'numpy',
'scipy',
'cython'
],
version = '0.2.2.25',
description = 'Hierarchical Poisson matrix factorization for recommender systems',
author = 'David Cortes',
author_email = 'david.cortes.rivera@gmail.com',
url = 'https://github.com/david-cortes/hpfrec',
keywords = ['poisson', 'probabilistic', 'non-negative', 'factorization', 'variational inference', 'collaborative filtering'],
classifiers = [],
cmdclass = {'build_ext': build_ext_subclass},
ext_modules = [Extension("hpfrec.cython_loops", sources=["hpfrec/cython_loops.pyx"], include_dirs=[numpy.get_include()])]
)
