cudnn-frontend/pyproject.toml at develop · liujane-dev/cudnn-frontend · 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
[build-system]
requires = ["setuptools>=64", "cmake>=3.18", "ninja==1.11.1.1", "pybind11[global]>=2.13,<3"]
build-backend = "setuptools.build_meta"

[project]
name = "nvidia-cudnn-frontend"
dynamic = ["version"]
description = "NVIDIA cuDNN Frontend — Python and C++ Graph API with SOTA attention (SDPA / Flash Attention), MoE grouped GEMM fusions, and FP8/MXFP8 kernels for Hopper and Blackwell GPUs."
readme = "README.md"
requires-python = ">=3.9"
license = {text = "MIT"}
keywords = [
    "cudnn",
    "cuda",
    "gpu",
    "nvidia",
    "deep-learning",
    "attention",
    "sdpa",
    "flash-attention",
    "transformer",
    "moe",
    "mixture-of-experts",
    "grouped-gemm",
    "fp8",
    "mxfp8",
    "blackwell",
    "hopper",
    "pytorch",
    "kernel",
    "graph-api",
]
classifiers = [
    "Development Status :: 5 - Production/Stable",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: MIT License",
    "Operating System :: POSIX :: Linux",
    "Operating System :: Microsoft :: Windows",
    "Programming Language :: C++",
    "Programming Language :: Python :: 3",
    "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",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Software Development :: Libraries :: Python Modules",
]

[tool.setuptools]
packages = {find = {where = ["python", "."], include = ["cudnn*", "include"], namespaces = true}}
package-dir = {"" = "python", "include" = "include"}
include-package-data = true

[project.urls]
"Homepage" = "https://github.com/NVIDIA/cudnn-frontend"
"Documentation" = "https://docs.nvidia.com/deeplearning/cudnn/frontend/latest/"
"Blog" = "https://nvidia.github.io/cudnn-frontend/"
"Repository" = "https://github.com/NVIDIA/cudnn-frontend"
"Bug Tracker" = "https://github.com/NVIDIA/cudnn-frontend/issues"
"Release Notes" = "https://github.com/NVIDIA/cudnn-frontend/releases"

[tool.setuptools.dynamic]
version = {attr = "cudnn.__version__"}

[tool.setuptools.package-data]
include = ["**/*"]

[project.optional-dependencies]
cutedsl = [
    "nvidia-cutlass-dsl[cu13]==4.5.0",
    "cuda-python",
    "torch",
    "apache-tvm-ffi",
    "torch-c-dlpack-ext",
]