Merge branch 'main' into rparolin/cuda.core.hashable

Author: rparolin

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()