<reponame>ChristianSteger/HORAYZON<filename>setup_cython.py<gh_stars>0
# Description: Setup file for Cython functions
#
# Building: python setup_cython.py build_ext --build-lib lib/
#
# Copyright (c) 2022 ETH Zurich, <NAME>
# MIT License

# Load modules
from distutils.core import setup
from Cython.Distutils import build_ext
from distutils.extension import Extension
import numpy

ext_modules = [Extension("functions_cy",
               ["src/functions_cy.pyx"],
               libraries=["m", "iomp5", "pthread"],
               extra_compile_args=["-O3", "-ffast-math", "-fopenmp"],
               extra_link_args=["-fopenmp"],
               include_dirs=[numpy.get_include()])]

setup(name="functions_cy",
      cmdclass={"build_ext": build_ext},
      ext_modules=ext_modules)
