from setuptools import setup
from pybind11.setup_helpers import Pybind11Extension, build_ext

native = "actioncodec2_kernels"
setup(
    ext_modules=[Pybind11Extension(
        "actioncodec2_kernels._actioncodec2_native",
        sources=[f"{native}/{name}.cpp" for name in ("module", "physical", "so3", "setbpe", "resample")],
        include_dirs=[native],
        depends=[f"{native}/parallel.hpp"],
        extra_compile_args=["-O3", "-fvisibility=hidden", "-ffp-contract=off"],
        cxx_std=17,
    )],
    cmdclass={"build_ext": build_ext},
)