import sys from setuptools import setup from pybind11.setup_helpers import Pybind11Extension, build_ext from pathlib import Path native = 'actioncodec2_kernels' openmp = ['-fopenmp'] if sys.platform.startswith('linux') else [] setup(name='actioncodec2-kernels', version='0.1.0', packages=['actioncodec2_kernels'], ext_modules=[Pybind11Extension('actioncodec2_kernels._native', sources=sorted(str(p) for p in Path(native).glob('*.cpp')), include_dirs=[native], cxx_std=17, extra_compile_args=['-O3', '-fvisibility=hidden', '-ffp-contract=off', *openmp], extra_link_args=openmp)], cmdclass={'build_ext': build_ext})