setup.py

# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE

import ast

from packaging.version import Version
from setuptools import setup
from setuptools_scm import get_version

version = get_version(
    root="..",
    relative_to=__file__,
    # We deliberately do not want to include the version suffixes (a/b/rc) in cuda-python versioning
    tag_regex="^(?P<version>v\\d+\\.\\d+\\.\\d+)",
    git_describe_command=["git", "describe", "--dirty", "--tags", "--long", "--match", "v*[0-9]*"],
)


version = Version(version).base_version


setup(
    version=version,
    install_requires=[
        f"cuda-bindings~={version}",
        "cuda-pathfinder~=1.1",
    ],
    extras_require={
        "all": [f"cuda-bindings[all]~={version}"],
    },
)