skip if ptx too new

Author: brandon-b-miller

cuda_core/cuda/core/experimental/_program.py

@@ -425,7 +425,8 @@ def close(self):
             self._linker.close()
         self._mnff.close()
 
-    def _can_load_generated_ptx(self):
+    @staticmethod
+    def _can_load_generated_ptx():
         driver_ver = handle_return(driver.cuDriverGetVersion())
         nvrtc_major, nvrtc_minor = handle_return(nvrtc.nvrtcVersion())
         return nvrtc_major * 1000 + nvrtc_minor * 10 <= driver_ver

cuda_core/tests/test_module.py

@@ -116,6 +116,8 @@ def test_object_code_load_ptx(get_saxpy_kernel_ptx):
     sym_map = mod._sym_map
     mod_obj = ObjectCode.from_ptx(ptx, symbol_mapping=sym_map)
     assert mod.code == ptx
+    if not Program._can_load_generated_ptx():
+        pytest.skip("PTX version too new for current driver")
     mod_obj.get_kernel("saxpy<double>")  # force loading