diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 1ac6754ae10..1e49cf1cbea 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -43,6 +43,14 @@ repos: exclude: '(.*pixi\.lock)|(\.git_archival\.txt)|(.*\.patch$)|(^cuda_core/cuda/core/_vendored/)' args: ["--fix"] + - id: check-pixi-cuda-version + name: Check pixi cuda-version pins track ci/versions.yml + entry: python ./ci/tools/check_pixi_cuda_version.py + language: python + additional_dependencies: [pyyaml] + files: '^(ci/versions\.yml|cuda_bindings/pixi\.toml|cuda_core/pixi\.toml)$' + pass_filenames: false + - id: no-markdown-in-docs-source name: Prevent markdown files in docs/source directories entry: bash -c diff --git a/ci/tools/check_pixi_cuda_version.py b/ci/tools/check_pixi_cuda_version.py new file mode 100644 index 00000000000..1ca931b9b48 --- /dev/null +++ b/ci/tools/check_pixi_cuda_version.py @@ -0,0 +1,91 @@ +# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-License-Identifier: Apache-2.0 + +"""Check pixi cuda-version pins track ci/versions.yml (cuda.build.version).""" + +from __future__ import annotations + +import sys +from pathlib import Path + +import tomllib +import yaml + +ROOT = Path(__file__).resolve().parents[2] +VERSIONS_FILE_PATH = ROOT / "ci" / "versions.yml" +PIXI_FILES = [ROOT / d / "pixi.toml" for d in ("cuda_bindings", "cuda_core")] + + +def main() -> int: + """Verify cuda_bindings/cuda_core pixi pins match ci/versions.yml.""" + if not VERSIONS_FILE_PATH.is_file(): + print(f"error: {VERSIONS_FILE_PATH} not found", file=sys.stderr) + return 2 + try: + build_version = yaml.safe_load(VERSIONS_FILE_PATH.read_text(encoding="utf-8"))["cuda"]["build"]["version"] + except (KeyError, TypeError): + print(f"error: cuda.build.version not found in {VERSIONS_FILE_PATH}", file=sys.stderr) + return 2 + + major, minor, *_ = build_version.split(".") + expected = f"{major}.{minor}.*" + cuda_feature = f"cu{major}" + + errors: list[str] = [] + checked: list[str] = [] + for path in PIXI_FILES: + if not path.is_file(): + print(f"error: {path} not found", file=sys.stderr) + return 2 + with path.open("rb") as f: + data = tomllib.load(f) + rel = path.relative_to(ROOT) + try: + variants = data["workspace"]["build-variants"]["cuda-version"] + cuda_pin = data["feature"][cuda_feature]["dependencies"]["cuda-version"] + except KeyError as exc: + print( + f"error: {rel} missing feature {cuda_feature!r} or cuda-version key: {exc}", + file=sys.stderr, + ) + return 2 + if expected not in variants: + errors.append( + f"{rel}: workspace.build-variants.cuda-version={variants!r} " + f"does not include {expected!r} " + f"(from ci/versions.yml cuda.build.version={build_version!r})" + ) + if cuda_pin != expected: + errors.append( + f"{rel}: feature.{cuda_feature}.dependencies.cuda-version={cuda_pin!r} " + f"!= {expected!r} " + f"(from ci/versions.yml cuda.build.version={build_version!r})" + ) + + checked.append( + f"{rel} (workspace.build-variants.cuda-version={variants!r}, " + f"feature.{cuda_feature}.dependencies.cuda-version={cuda_pin!r})" + ) + + if errors: + print( + f"error: cuda_bindings/cuda_core pixi cuda-version pins out of sync with " + f"ci/versions.yml cuda.build.version={build_version!r} " + f"(expected pin {expected!r}):", + file=sys.stderr, + ) + for err in errors: + print(f" - {err}", file=sys.stderr) + return 1 + + print( + f"OK: pixi cuda-version pins match ci/versions.yml " + f"cuda.build.version={build_version!r} (expected pin {expected!r}):" + ) + for item in checked: + print(f" - {item}") + return 0 + + +if __name__ == "__main__": + sys.exit(main())