# 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__,
# Preserve a/b pre-release suffixes, but intentionally strip rc suffixes.
tag_regex="^(?Pv\\d+\\.\\d+\\.\\d+(?:[ab]\\d+)?)",
git_describe_command=["git", "describe", "--dirty", "--tags", "--long", "--match", "v*[0-9]*"],
)
base_version = Version(version).base_version
if base_version == version:
# Tagged release
matcher = "~="
else:
# Pre-release version
matcher = "=="
setup(
version=version,
install_requires=[
f"cuda-bindings{matcher}{version}",
"cuda-core~=1.0.0",
"cuda-pathfinder~=1.1",
],
extras_require={
"all": [f"cuda-bindings[all]{matcher}{version}"],
},
)