pyproject.toml

# This file configures wheels compilation for `cibuilwheel` for NumKong CPython bindings.
# On a good day it will produce:
#   - `manylinux` and `musllinux` vectorized wheels for Linux on x86_64, aarch64;
#   - `manylinux` and `musllinux` wheels for Linux on i686, ppc64le, s390x;
#   - `macos` wheels for x86_64, arm64;
#   - `windows` wheels for AMD64, ARM64.
#   * for 5 Python versions from 3.10 to 3.14.
#   * for 2 free-threading Python versions from 3.13 to 3.14.
#   * running thousands of fuzzy tests on each wheel.
#   = meaning 12 platforms * 7 Python versions = 84 builds.
[build-system]
build-backend = "setuptools.build_meta"
requires = ["setuptools>=42"]

[tool.pytest.ini_options]
addopts = ["-ra", "--showlocals", "--strict-markers", "--strict-config"]
filterwarnings = ["error"]
minversion = "6.0"
xfail_strict = true

[tool.cibuildwheel]
build-verbosity = 0
test-requires = ["pytest", "pytest-repeat", "pytest-randomly", "tabulate"]

# Assuming NumPy and SciPy aren't available precompiled for some platforms,
# we will end up compiling them from source on emulated hardware...
# To avoid that use `--only-binary=:all:` to only fetch precompiled wheels.
# For better compatibility, we use (&&) to chain commands instead of (;).
test-command = """
    python -m pip install numpy --only-binary=:all: || (echo "Failed to install NumPy" && exit 0) &&
    python -m pip install scipy --only-binary=:all: || (echo "Failed to install SciPy" && exit 0) &&
    python -m pip install ml_dtypes --only-binary=:all: || (echo "Failed to install ml_dtypes" && exit 0) &&
    python -c "import numkong; print(numkong.get_capabilities())" &&
    PYTHONMALLOC=debug pytest {project}/test/ -s -x -Wd
"""

# We need to build for all platforms:
# - on Linux: x86_64, aarch64, i686
# - on macOS: x86_64, arm64
# - on Windows: AMD64, ARM64
# https://cibuildwheel.readthedocs.io/en/stable/options/#archs
#
# Important to note, not all those platforms have recent images.
# The `manylinux_2_28` seems to be missing for `i686`.
# The `i686` is 32-bit x86, and `x86_64` is 64-bit x86.
archs = ["all"]

# Add "pp*" to skip PyPy builds, but they should work fine these days :)
# https://cibuildwheel.readthedocs.io/en/stable/options/#build-skip
# https://cibuildwheel.readthedocs.io/en/stable/#what-does-it-do
enable = [] # don't enable PyPy
skip = [] # we also don't use any SIMD on: ["*s390x*", "*ppc64le*"]

# Testing half-precision math in QEMU is unreliable!
environment-pass = [
  "NK_IN_QEMU",
  "NK_SEED",
  "NK_REPETITIONS",
  "NK_DENSE_DIMENSIONS",
  "NK_CURVED_DIMENSIONS",
  "NK_MATRIX_HEIGHT",
  "NK_MATRIX_WIDTH",
  "NK_MATRIX_DEPTH",
]

[tool.cibuildwheel.environment]
NK_BUILD_PARALLEL = "2"
NK_IN_QEMU = "1"

[tool.cibuildwheel.linux]
before-build = ["rm -rf {project}/build"]
repair-wheel-command = "auditwheel repair -w {dest_dir} {wheel}"

# Use more recent images for the most popular SIMD-capable CPU architectures, to have access to newer compilers.
# Otherwise, prepare yourself to all kinds of AVX-512 issues and other SIMD-related pain.
# You can keep track of the most recent images on Quay:
# - for `manylinux`: https://quay.io/search?q=manylinux
# - for `musllinux`: https://quay.io/search?q=musllinux
manylinux-aarch64-image = "manylinux_2_28"
manylinux-x86_64-image = "manylinux_2_28"
musllinux-aarch64-image = "musllinux_1_2"
musllinux-x86_64-image = "musllinux_1_2"
# RISC-V wheel uploads are temporarily disabled.
# The official PyPA riscv64 images currently provide LLVM 20 packages, while the
# portable `rv64gcv` baseline plus richer RISC-V kernels now requires Clang 23.
# Keep the image names here as comments until we have either custom images or a
# packaged LLVM 23 path for both manylinux and musllinux.
# manylinux-riscv64-image = "manylinux_2_39"
# musllinux-riscv64-image = "musllinux_1_2"

# On CentOS we have to use `yum`.
# The healthy version would be: `apt-get update && apt-get install -y libc6-dev wget python3-dev`.
# We can't even use direct YUM variant: `yum update -y && yum install -y glibc-devel wget python3-devel`
# as downloading from `extras` fails on Arm instances!
before-all = [
  "yum update -y --disablerepo=extras --setopt=install_weak_deps=0 && yum install -y --disablerepo=extras glibc-devel wget python3-devel",
]

# With `musl` builds, we obviously don't need the `glibc` and can't use `yum`.
# This may also be handy for using custom dependencies for different Python versions:
# https://cibuildwheel.readthedocs.io/en/stable/options/#overrides
[[tool.cibuildwheel.overrides]]
before-all = "apk add --update wget python3-dev"
select = "*-musllinux*"

[tool.cibuildwheel.macos]
before-build = ["rm -rf {project}/build"]
repair-wheel-command = "delocate-wheel --require-archs {delocate_archs} -w {dest_dir} -v {wheel}"

[tool.cibuildwheel.windows]
before-build = ["rd /s /q {project}\\build || echo Done"]

# Windows-based OSes and PowerShell don't support the logical operators
# In PowerShell we could write something like:
#   python -m pip install numpy; if ($LASTEXITCODE -ne 0) { echo "Failed to install NumPy... Nothing can stop me, I'm all the way up!" }
# but its safer to use the CMD syntax
[[tool.cibuildwheel.overrides]]
select = "*win*"
test-command = """
    cmd /c "python -m pip install numpy & if errorlevel 1 echo Failed to install NumPy... Nothing can stop me, I'm all the way up! &
    python -m pip install scipy & if errorlevel 1 echo Failed to install SciPy... Nothing can stop me, I'm all the way up! &
    python -m pip install ml_dtypes & if errorlevel 1 echo Failed to install ml_dtypes... Nothing can stop me, I'm all the way up! &
    set PYTHONMALLOC=debug && pytest {project}/test/ -s -x -Wd"
"""

# These environment variables allow setup.py to detect the target architecture
# during cross-compilation (e.g., building ARM64 wheels on x64 host).
# The NK_TARGET_* variables override platform.machine() detection.

# x86-64 builds
[[tool.cibuildwheel.overrides]]
environment.NK_TARGET_ARM_ = "0"
environment.NK_TARGET_X86_ = "1"
inherit.environment = "append"
select = "*-win_amd64"

[[tool.cibuildwheel.overrides]]
environment.NK_TARGET_ARM_ = "0"
environment.NK_TARGET_X86_ = "1"
inherit.environment = "append"
select = "*-manylinux*_x86_64"

[[tool.cibuildwheel.overrides]]
environment.NK_TARGET_ARM_ = "0"
environment.NK_TARGET_X86_ = "1"
inherit.environment = "append"
select = "*-musllinux*_x86_64"

[[tool.cibuildwheel.overrides]]
environment.NK_TARGET_ARM_ = "0"
environment.NK_TARGET_X86_ = "1"
inherit.environment = "append"
select = "*-macos*_x86_64"

# ARM64 builds
[[tool.cibuildwheel.overrides]]
environment.NK_TARGET_ARM_ = "1"
environment.NK_TARGET_X86_ = "0"
inherit.environment = "append"
select = "*-win_arm64"

[[tool.cibuildwheel.overrides]]
environment.NK_TARGET_ARM_ = "1"
environment.NK_TARGET_X86_ = "0"
inherit.environment = "append"
select = "*-manylinux*_aarch64"

[[tool.cibuildwheel.overrides]]
environment.NK_TARGET_ARM_ = "1"
environment.NK_TARGET_X86_ = "0"
inherit.environment = "append"
select = "*-musllinux*_aarch64"

[[tool.cibuildwheel.overrides]]
environment.NK_TARGET_ARM_ = "1"
environment.NK_TARGET_X86_ = "0"
inherit.environment = "append"
select = "*-macos*_arm64"

# RISC-V 64-bit wheels are temporarily disabled.
# Keep the old overrides commented out as a breadcrumb for re-enabling the
# release path once an official LLVM 23 package source is available inside the
# PyPA riscv64 images, or once we switch to custom images with LLVM 23 baked in.
#
# [[tool.cibuildwheel.overrides]]
# before-all = "dnf update -y && dnf install -y glibc-devel wget python3-devel clang lld"
# environment.CC = "clang"
# environment.CXX = "clang++"
# environment.LDFLAGS = "-fuse-ld=lld"
# environment.NK_TARGET_ARM_ = "0"
# environment.NK_TARGET_RISCV_ = "1"
# environment.NK_TARGET_X86_ = "0"
# inherit.environment = "append"
# select = "*-manylinux*_riscv64"
#
# [[tool.cibuildwheel.overrides]]
# before-all = "apk add --update wget python3-dev clang lld"
# environment.CC = "clang"
# environment.CXX = "clang++"
# environment.LDFLAGS = "-fuse-ld=lld"
# environment.NK_TARGET_ARM_ = "0"
# environment.NK_TARGET_RISCV_ = "1"
# environment.NK_TARGET_X86_ = "0"
# inherit.environment = "append"
# select = "*-musllinux*_riscv64"

# Pyodide / Emscripten WASM builds
# The Pyodide venv creates Python-script entry points (like `pytest`) that can't
# be executed directly as shell commands, so we must use `python -m pytest`.
# Also, `ml_dtypes` has no Pyodide wheel, so we skip it.
[[tool.cibuildwheel.overrides]]
before-build = ["rm -rf {project}/build"]
select = "*-pyodide*"
test-command = """
    python -m pip install numpy --only-binary=:all: || (echo "Failed to install NumPy" && exit 0) &&
    python -m pip install scipy --only-binary=:all: || (echo "Failed to install SciPy" && exit 0) &&
    python -c "import numkong; print(numkong.get_capabilities())" &&
    PYTHONMALLOC=debug python -m pytest {project}/test/ -s -x -Wd
"""

[tool.ruff]
line-length = 120
target-version = "py310"

[tool.ruff.lint]
ignore = [
  "E712", # == True/False: intentional for testing scalar __bool__ return values
  "E731", # lambda assignments: clearer for one-liner baselines
  "B905", # zip-without-strict: not safety-critical in test code
  "SIM105", # contextlib.suppress: explicit try/except is clearer here
  "B028", # stacklevel in warnings.warn: not important in test code
  "RUF002", # ambiguous unicode in docstrings: intentional math symbols
  "RUF003", # ambiguous unicode in comments: intentional math symbols
  "RUF043", # pytest.raises regex patterns: the metacharacters are intentional
]
select = ["E", "F", "W", "I", "UP", "B", "SIM", "RUF"]

[tool.ruff.lint.per-file-ignores]
"test/*" = ["E501", "F811"]

# Configuration options for the Black formatter:
# https://black.readthedocs.io/en/latest/usage_and_configuration/the_basics.html#where-black-looks-for-the-file
[tool.black]
line-length = 120 # Set line length to the same value as in `.clang-format` for modern wide screens
target-version = ['py310', 'py314']