diff --git a/cuda_core/tests/test_program.py b/cuda_core/tests/test_program.py index 0b23bc1aa07..adc778973c2 100644 --- a/cuda_core/tests/test_program.py +++ b/cuda_core/tests/test_program.py @@ -2,13 +2,14 @@ # # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE +import re import warnings import pytest from cuda.core.experimental import _linker from cuda.core.experimental._module import Kernel, ObjectCode from cuda.core.experimental._program import Program, ProgramOptions -from cuda.core.experimental._utils.cuda_utils import driver, handle_return +from cuda.core.experimental._utils.cuda_utils import CUDAError, driver, handle_return cuda_driver_version = handle_return(driver.cuDriverGetVersion()) is_culink_backend = _linker._decide_nvjitlink_or_driver() @@ -317,7 +318,12 @@ def test_nvvm_program_creation_compilation(nvvm_ir): assert program.backend == "NVVM" assert program.handle is not None obj = program.compile("ptx") - ker = obj.get_kernel("simple") # noqa: F841 + try: + ker = obj.get_kernel("simple") # noqa: F841 + except CUDAError as e: + if re.search(r"CUDA_UNSUPPORTED_PTX_VERSION", str(e)): + pytest.xfail("PTX version not supported by current CUDA Driver") + raise program.close()