From 4403b0a9e7fdee6cbc3ce7f3eee3237ac5b2b56e Mon Sep 17 00:00:00 2001 From: abhilash1910 Date: Fri, 31 Oct 2025 06:18:50 +0000 Subject: [PATCH 1/3] add exception for mismatching PTX and supporting CUDA Driver versions --- cuda_core/tests/test_program.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/cuda_core/tests/test_program.py b/cuda_core/tests/test_program.py index 0b23bc1aa07..63153e583ff 100644 --- a/cuda_core/tests/test_program.py +++ b/cuda_core/tests/test_program.py @@ -5,10 +5,11 @@ import warnings import pytest +import re 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 driver, handle_return, CUDAError cuda_driver_version = handle_return(driver.cuDriverGetVersion()) is_culink_backend = _linker._decide_nvjitlink_or_driver() @@ -317,7 +318,11 @@ 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") program.close() From 05cac1216d0d575e7236611e89156b3565b2f2a2 Mon Sep 17 00:00:00 2001 From: abhilash1910 Date: Fri, 31 Oct 2025 06:24:45 +0000 Subject: [PATCH 2/3] add exception for mismatching PTX and supporting CUDA Driver versions --- cuda_core/tests/test_program.py | 1 + 1 file changed, 1 insertion(+) diff --git a/cuda_core/tests/test_program.py b/cuda_core/tests/test_program.py index 63153e583ff..8bbfdf2c1e6 100644 --- a/cuda_core/tests/test_program.py +++ b/cuda_core/tests/test_program.py @@ -323,6 +323,7 @@ def test_nvvm_program_creation_compilation(nvvm_ir): 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() From dd78838afab3e74720bf7027630515405926d434 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 31 Oct 2025 06:31:55 +0000 Subject: [PATCH 3/3] [pre-commit.ci] auto code formatting --- cuda_core/tests/test_program.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cuda_core/tests/test_program.py b/cuda_core/tests/test_program.py index 8bbfdf2c1e6..adc778973c2 100644 --- a/cuda_core/tests/test_program.py +++ b/cuda_core/tests/test_program.py @@ -2,14 +2,14 @@ # # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE +import re import warnings import pytest -import re 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, CUDAError +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()