cuda-python/cuda_bindings/pyproject.toml at d76e851c4ead13c84e973fa9ce372277f8b3d5ff · NVIDIA/cuda-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
# SPDX-FileCopyrightText: Copyright (c) 2023-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
[build-system]
requires = [
    "setuptools>=80.0.0",
    "setuptools_scm[simple]>=8,!=10.1",
    "cython>=3.2,<3.3",
    "pyclibrary>=0.1.7",
    "cuda-pathfinder>=1.5",
]
build-backend = "build_hooks"
backend-path = ["."]

[project]
name = "cuda-bindings"
description = "Python bindings for CUDA"
authors = [
    { name = "NVIDIA Corporation", email = "cuda-python-conduct@nvidia.com" },
]
license = "LicenseRef-NVIDIA-SOFTWARE-LICENSE"
requires-python = ">=3.10"
classifiers = [
    "Intended Audience :: Developers",
    "Topic :: Database",
    "Topic :: Scientific/Engineering",
    "Programming Language :: Python",
    "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",
    "Environment :: GPU :: NVIDIA CUDA",
]
dynamic = ["version", "readme"]
dependencies = ["cuda-pathfinder >=1.4.2"]

[project.optional-dependencies]
all = [
    "cuda-toolkit[nvrtc,nvjitlink,nvvm,nvfatbin]==13.*",
    "cuda-toolkit[cufile]==13.*; sys_platform == 'linux'",
    "cuda-toolkit==13.*",
    "nvidia-cudla==13.*; platform_system == 'Linux' and platform_machine == 'aarch64'",
]

[dependency-groups]
test = [
    "cython>=3.2,<3.3",
    "setuptools>=80.0.0",
    # TODO: remove the Python 3.15 guard once 3.15 is officially supported
    "matplotlib>=3.5.0,<=3.10.9; python_version < '3.15'",
    "numpy>=1.21.1,<=2.5.0",
    "pytest==9.1.1",
    "pytest-benchmark==5.2.3",
    "pytest-repeat==0.9.4",
    "pytest-randomly==4.1.0",
    "pyglet==2.1.14",
]

[project.urls]
Repository = "https://github.com/NVIDIA/cuda-python"
Documentation = "https://nvidia.github.io/cuda-python/"

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

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

[tool.setuptools.package-data]
"*" = ["*.pxd", "*.pxi"]
"cuda.bindings._bindings" = ["*.h"]
"cuda.bindings._lib" = ["*.h"]

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

[tool.cibuildwheel]
skip = "*-musllinux_*"
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}"

[tool.pytest.ini_options]
required_plugins = "pytest-benchmark"
addopts = "--benchmark-disable --showlocals"
norecursedirs = ["tests/cython", "examples"]
xfail_strict = true
# Keep this authorship marker registry in sync across all pytest config roots.
# Search for "agent_authored(model)" before editing.
markers = [
    "agent_authored(model): agent-authored test not yet materially human-reviewed",
    "human_reviewed: agent-authored test materially reviewed or rewritten by a human",
    "human_authored: test authored primarily by a human",
]

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