# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
#
# Use of this software is governed by the terms and conditions of the
# NVIDIA End User License Agreement (EULA), available at:
# https://docs.nvidia.com/cutlass/media/docs/pythonDSL/license.html
#
# Any use, reproduction, disclosure, or distribution of this software
# and related documentation outside the scope permitted by the EULA
# is strictly prohibited.

from typing import Callable

from .common import DSLCudaVersion, DSLRuntimeError, _get_cuda_version

try:
    CUDA_VERSION = DSLCudaVersion(_get_cuda_version())
except Exception as e:
    raise DSLRuntimeError(
        "💥💥💥 Failed to get CUDA version 💥💥💥",
        cause=e,
        suggestion="Consider re-installing the package.",
    ) from e