setup.py

from setuptools import setup, Extension
import platform
import os
import sys

compile_args = ["-std=c++17", "-O3", "-ffast-math"]
if platform.system() != "Windows":
    archflags = os.environ.get("ARCHFLAGS", "")
    if "x86_64" in archflags:
        compile_args.append("-march=x86-64")
    elif "arm64" in archflags:
        compile_args.append("-mcpu=apple-m1")
    elif platform.machine() == "x86_64":
        compile_args.append("-march=x86-64")
    elif platform.machine() == "aarch64" or platform.machine() == "arm64":
        compile_args.append("-mcpu=native")

if sys.platform == "win32":
    compile_args = ["/std:c++17", "/O2", "/fp:fast"]

setup(
    ext_modules=[
        Extension(
            "kelzer",
            sources=["cpp/kelzer.cpp"],
            language="c++",
            extra_compile_args=compile_args,
        )
    ],
)