Got initial version to address issue 2183. Let pre-commit check covers pixi cuda version pins to ci/versions.yml

Author: lijinf2

.pre-commit-config.yaml

@@ -50,6 +50,14 @@ repos:
         files: ^cuda_bindings/
         types: [text]
 
+      - 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

ci/tools/check_pixi_cuda_version.py

@@ -0,0 +1,61 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+"""Check pixi cu13 cuda-version pins track ci/versions.yml (cuda.build.version)."""
+
+from __future__ import annotations
+
+import sys
+import tomllib
+from pathlib import Path
+
+import yaml
+
+ROOT = Path(__file__).resolve().parents[2]
+VERSIONS = ROOT / "ci" / "versions.yml"
+PIXI_FILES = [ROOT / d / "pixi.toml" for d in ("cuda_bindings", "cuda_core")]
+
+
+def main() -> int:
+    if not VERSIONS.is_file():
+        print(f"error: {VERSIONS} not found", file=sys.stderr)
+        return 2
+    try:
+        build_version = yaml.safe_load(VERSIONS.read_text(encoding="utf-8"))["cuda"]["build"]["version"]
+    except (KeyError, TypeError):
+        print(f"error: cuda.build.version not found in {VERSIONS}", file=sys.stderr)
+        return 2
+
+    major, minor, *_ = build_version.split(".")
+    expected = f"{major}.{minor}.*"
+    errors: 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"]
+            cu13 = data["feature"]["cu13"]["dependencies"]["cuda-version"]
+        except KeyError as exc:
+            print(f"error: {rel} missing cuda-version key: {exc}", file=sys.stderr)
+            return 2
+        if expected not in variants:
+            errors.append(f"{rel}: build-variants missing {expected!r} (has {variants})")
+        if cu13 != expected:
+            errors.append(f"{rel}: cu13 pin is {cu13!r}, expected {expected!r}")
+
+    if errors:
+        print(f"error: pixi cuda-version pins out of sync (expected {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 ({expected!r})")
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())