Add version validation to prevent silent setuptools-scm fallback versions

cuda_bindings/cuda/bindings/__init__.py

@@ -1,5 +1,38 @@
 # SPDX-FileCopyrightText: Copyright (c) 2021-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
 
+import os
+
 from cuda.bindings import utils
 from cuda.bindings._version import __version__
+
+# Version validation: detect setuptools-scm fallback versions (e.g., 0.1.dev...)
+# This check must be kept in sync with similar checks in cuda.core and cuda.pathfinder
+if not os.environ.get("CUDA_PYTHON_ALLOW_FALLBACK_VERSIONING"):
+    version_parts = __version__.split(".")
+    if len(version_parts) < 2:
+        raise RuntimeError(
+            f"Invalid version format: '{__version__}'. "
+            f"The version detection system failed. "
+            f"This usually means git tags are not available (e.g., shallow clone or zip archive). "
+            f"To fix: ensure you have a full git checkout with tags, or set "
+            f"CUDA_PYTHON_ALLOW_FALLBACK_VERSIONING=1 to disable this check."
+        )
+    try:
+        major, minor = int(version_parts[0]), int(version_parts[1])
+    except ValueError:
+        raise RuntimeError(
+            f"Invalid version format: '{__version__}'. "
+            f"The version detection system failed. "
+            f"This usually means git tags are not available (e.g., shallow clone or zip archive). "
+            f"To fix: ensure you have a full git checkout with tags, or set "
+            f"CUDA_PYTHON_ALLOW_FALLBACK_VERSIONING=1 to disable this check."
+        ) from None
+    if major == 0 and minor <= 1:
+        raise RuntimeError(
+            f"Invalid version detected: '{__version__}'. "
+            f"The version detection system failed silently and produced a fallback version. "
+            f"This usually means git tags are not available (e.g., shallow clone or zip archive). "
+            f"To fix: ensure you have a full git checkout with tags, or set "
+            f"CUDA_PYTHON_ALLOW_FALLBACK_VERSIONING=1 to disable this check."
+        )

cuda_core/cuda/core/__init__.py

@@ -2,8 +2,41 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+import os
+
 from cuda.core._version import __version__
 
+# Version validation: detect setuptools-scm fallback versions (e.g., 0.1.dev...)
+# This check must be kept in sync with similar checks in cuda.bindings and cuda.pathfinder
+if not os.environ.get("CUDA_PYTHON_ALLOW_FALLBACK_VERSIONING"):
+    version_parts = __version__.split(".")
+    if len(version_parts) < 2:
+        raise RuntimeError(
+            f"Invalid version format: '{__version__}'. "
+            f"The version detection system failed. "
+            f"This usually means git tags are not available (e.g., shallow clone or zip archive). "
+            f"To fix: ensure you have a full git checkout with tags, or set "
+            f"CUDA_PYTHON_ALLOW_FALLBACK_VERSIONING=1 to disable this check."
+        )
+    try:
+        major, minor = int(version_parts[0]), int(version_parts[1])
+    except ValueError:
+        raise RuntimeError(
+            f"Invalid version format: '{__version__}'. "
+            f"The version detection system failed. "
+            f"This usually means git tags are not available (e.g., shallow clone or zip archive). "
+            f"To fix: ensure you have a full git checkout with tags, or set "
+            f"CUDA_PYTHON_ALLOW_FALLBACK_VERSIONING=1 to disable this check."
+        ) from None
+    if major == 0 and minor <= 1:
+        raise RuntimeError(
+            f"Invalid version detected: '{__version__}'. "
+            f"The version detection system failed silently and produced a fallback version. "
+            f"This usually means git tags are not available (e.g., shallow clone or zip archive). "
+            f"To fix: ensure you have a full git checkout with tags, or set "
+            f"CUDA_PYTHON_ALLOW_FALLBACK_VERSIONING=1 to disable this check."
+        )
+
 try:
     from cuda import bindings
 except ImportError:

cuda_pathfinder/cuda/pathfinder/__init__.py

@@ -3,6 +3,8 @@
 
 """cuda.pathfinder public APIs"""
 
+import os
+
 from cuda.pathfinder._dynamic_libs.load_dl_common import DynamicLibNotFoundError as DynamicLibNotFoundError
 from cuda.pathfinder._dynamic_libs.load_dl_common import LoadedDL as LoadedDL
 from cuda.pathfinder._dynamic_libs.load_nvidia_dynamic_lib import load_nvidia_dynamic_lib as load_nvidia_dynamic_lib
@@ -14,6 +16,37 @@
 
 from cuda.pathfinder._version import __version__  # isort: skip  # noqa: F401
 
+# Version validation: detect setuptools-scm fallback versions (e.g., 0.1.dev...)
+# This check must be kept in sync with similar checks in cuda.bindings and cuda.core
+if not os.environ.get("CUDA_PYTHON_ALLOW_FALLBACK_VERSIONING"):
+    version_parts = __version__.split(".")
+    if len(version_parts) < 2:
+        raise RuntimeError(
+            f"Invalid version format: '{__version__}'. "
+            f"The version detection system failed. "
+            f"This usually means git tags are not available (e.g., shallow clone or zip archive). "
+            f"To fix: ensure you have a full git checkout with tags, or set "
+            f"CUDA_PYTHON_ALLOW_FALLBACK_VERSIONING=1 to disable this check."
+        )
+    try:
+        major, minor = int(version_parts[0]), int(version_parts[1])
+    except ValueError:
+        raise RuntimeError(
+            f"Invalid version format: '{__version__}'. "
+            f"The version detection system failed. "
+            f"This usually means git tags are not available (e.g., shallow clone or zip archive). "
+            f"To fix: ensure you have a full git checkout with tags, or set "
+            f"CUDA_PYTHON_ALLOW_FALLBACK_VERSIONING=1 to disable this check."
+        ) from None
+    if major == 0 and minor <= 1:
+        raise RuntimeError(
+            f"Invalid version detected: '{__version__}'. "
+            f"The version detection system failed silently and produced a fallback version. "
+            f"This usually means git tags are not available (e.g., shallow clone or zip archive). "
+            f"To fix: ensure you have a full git checkout with tags, or set "
+            f"CUDA_PYTHON_ALLOW_FALLBACK_VERSIONING=1 to disable this check."
+        )
+
 # Indirections to help Sphinx find the docstrings.
 #: Mapping from short CUDA Toolkit (CTK) library names to their canonical
 #: header basenames (used to validate a discovered include directory).