pyproject.toml

# SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# SPDX-License-Identifier: Apache-2.0

[build-system]
requires = [
    "setuptools>=80",
    "setuptools-scm[simple]>=8",
    "Cython>=3.2,<3.3",
]
build-backend = "build_hooks"
backend-path = ["."]


[project]
name = "cuda-core"
dynamic = [
    "version",
    "readme",
]
requires-python = '>=3.10'
description = "cuda.core: pythonic CUDA module"
authors = [
    { name = "NVIDIA Corporation" }
]
license = "Apache-2.0"
classifiers = [
    "Development Status :: 4 - Beta",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "Intended Audience :: End Users/Desktop",
    "Natural Language :: English",
    "Operating System :: POSIX :: Linux",
    "Operating System :: Microsoft :: Windows",
    "Topic :: Education",
    "Topic :: Scientific/Engineering",
    "Topic :: Software Development :: Libraries",
    "Programming Language :: Python :: 3 :: Only",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: 3.13",
    "Programming Language :: Python :: 3.14",
    "Programming Language :: Python :: Implementation :: CPython",
    "Environment :: GPU :: NVIDIA CUDA",
    "Environment :: GPU :: NVIDIA CUDA :: 12",
    "Environment :: GPU :: NVIDIA CUDA :: 13",
]
dependencies = [
    "cuda-pathfinder >=1.4.2",
    "numpy",
]

[project.optional-dependencies]
cu12 = ["cuda-bindings[all]==12.*"]
cu13 = ["cuda-bindings[all]==13.*"]

[dependency-groups]
test = ["cython>=3.2,<3.3", "setuptools", "pytest>=6.2.4", "pytest-benchmark", "pytest-randomly", "pytest-repeat", "pytest-rerunfailures", "cloudpickle", "psutil", "cffi"]
ml-dtypes = ["ml-dtypes>=0.5.4,<0.6.0"]
test-cu12 = [
    {include-group = "ml-dtypes" },
    {include-group = "test" },
    "cupy-cuda12x; python_version < '3.14'",
    "cuda-toolkit[cudart]==12.*,!=12.9.2; sys_platform == 'win32'",
    "cuda-toolkit[cudart]==12.*; sys_platform != 'win32'",
]  # runtime headers needed by CuPy
test-cu13 = [
    {include-group = "ml-dtypes" },
    {include-group = "test" },
    "cupy-cuda13x; python_version < '3.14'",
    "cuda-toolkit[cudart]==13.*,!=13.0.3; sys_platform == 'win32'",
    "cuda-toolkit[cudart]==13.*; sys_platform != 'win32'",
]  # runtime headers needed by CuPy
# free threaded build, cupy doesn't support free-threaded builds yet, so avoid installing it for now
# TODO: cupy should support free threaded builds
test-cu12-ft = [
    {include-group = "ml-dtypes" },
    {include-group = "test" },
    "cuda-toolkit[cudart]==12.*,!=12.9.2; sys_platform == 'win32'",
    "cuda-toolkit[cudart]==12.*; sys_platform != 'win32'",
]
test-cu13-ft = [
    {include-group = "ml-dtypes" },
    {include-group = "test" },
    "cuda-toolkit[cudart]==13.*,!=13.0.3; sys_platform == 'win32'",
    "cuda-toolkit[cudart]==13.*; sys_platform != 'win32'",
]

[tool.uv]
conflicts = [
    [
        { extra = "cu12" },
        { extra = "cu13" },
    ],
    [
        { group = "test-cu12" },
        { group = "test-cu13" },
        { group = "test-cu12-ft" },
        { group = "test-cu13-ft" },
    ],
]

[project.urls]
homepage = "https://nvidia.github.io/cuda-python/"
documentation = "https://nvidia.github.io/cuda-python/cuda-core/"
repository = "https://github.com/NVIDIA/cuda-python/tree/main/cuda_core"
issues = "https://github.com/NVIDIA/cuda-python/issues/"

[tool.setuptools.packages.find]
include = ["cuda.core*"]

[tool.setuptools]
include-package-data = false

[tool.setuptools.package-data]
"*" = ["*.pxd"]
"cuda.core._include" = ["*.h", "*.hpp"]
"cuda.core._cpp" = ["*.hpp"]

[tool.setuptools.dynamic]
readme = { file = ["DESCRIPTION.rst"], content-type = "text/x-rst" }

[tool.setuptools_scm]
root = ".."
version_file = "cuda/core/_version.py"
# Preserve a/b pre-release suffixes, but intentionally strip rc suffixes.
tag_regex = "^cuda-core-(?P<version>v\\d+\\.\\d+\\.\\d+(?:[ab]\\d+)?)"
git_describe_command = ["git", "describe", "--dirty", "--tags", "--long", "--match", "cuda-core-v*[0-9]*"]

[tool.cibuildwheel]
skip = "*-musllinux_*"
enable = "cpython-freethreading"
build-verbosity = 1

[tool.cibuildwheel.linux]
archs = "native"

[tool.cibuildwheel.windows]
archs = "AMD64"
before-build = "pip install delvewheel"
repair-wheel-command = "delvewheel repair --namespace-pkg cuda -w {dest_dir} {wheel}"