setup_cpp.py

"""Build the quoridor_cpp extension:  python setup_cpp.py build_ext --inplace"""
import sys

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

# -O3 is a GCC/Clang flag; MSVC (Windows) doesn't recognize it (silently
# ignored with a D9002 warning, or errors), which would leave the C++ engine
# built WITHOUT optimization -- catastrophic for the BFS/MCTS-heavy engine
# code even though it has no effect on PyTorch/CUDA's own eval throughput.
extra_compile_args = ["/O2"] if sys.platform == "win32" else ["-O3"]
extra_link_args = []

# TEMP DIAGNOSTIC: AddressSanitizer build to locate a Windows-only access
# violation. Remove before shipping (adds overhead, changes runtime behavior).
# /Od (instead of /O2) + /DEBUG keep function boundaries intact and emit a
# real PDB so llvm-symbolizer can resolve real function names/lines instead
# of collapsing everything into the nearest exported symbol (PyInit_*+offset).
import os
if sys.platform == "win32" and os.environ.get("QUORIDOR_ASAN") == "1":
    extra_compile_args = ["/Od", "/Zi", "/fsanitize=address"]
    extra_link_args += ["/DEBUG"]

setup(
    name="quoridor_cpp",
    version="0.1.0",
    ext_modules=[
        Pybind11Extension(
            "quoridor_cpp",
            ["cpp/bindings.cpp"],
            # Without this, setuptools only stats bindings.cpp and skips the
            # rebuild when a header-only change is all that happened.
            depends=[
                "cpp/engine.hpp",
                "cpp/selfplay.hpp",
                "cpp/tournament.hpp",
                "cpp/alphabeta.hpp",
                "cpp/rollout_mcts.hpp",
            ],
            cxx_std=17,
            extra_compile_args=extra_compile_args,
            extra_link_args=extra_link_args,
        )
    ],
    cmdclass={"build_ext": build_ext},
)