For drivers with version in [12000, 12040) skip tests requiring KernelParamInfo

Except for one test where we check that NotImplementedError is raised.

Author: oleksandr-pavlyk

cuda_core/tests/test_module.py

@@ -32,12 +32,12 @@
 
 
 @pytest.fixture(scope="module")
-def cuda12_prerequisite_check():
+def cuda12_4_prerequisite_check():
     # binding availability depends on cuda-python version
     # and version of underlying CUDA toolkit
     _py_major_ver, _ = get_binding_version()
     _driver_ver = handle_return(driver.cuDriverGetVersion())
-    return _py_major_ver >= 12 and _driver_ver >= 12000
+    return _py_major_ver >= 12 and _driver_ver >= 12040
 
 
 def test_kernel_attributes_init_disabled():
@@ -180,12 +180,15 @@ def test_object_code_handle(get_saxpy_object_code):
     assert mod.handle is not None
 
 
-def test_saxpy_arguments(get_saxpy_kernel, cuda12_prerequisite_check):
-    if not cuda12_prerequisite_check:
-        pytest.skip("Test requires CUDA 12")
+def test_saxpy_arguments(get_saxpy_kernel, cuda12_4_prerequisite_check):
     krn, _ = get_saxpy_kernel
 
-    assert krn.num_arguments == 5
+    if cuda12_4_prerequisite_check:
+        assert krn.num_arguments == 5
+    else:
+        with pytest.raises(NotImplementedError):
+            _ = krn.num_arguments
+        return
 
     assert "ParamInfo" in str(type(krn).arguments_info.fget.__annotations__)
     arg_info = krn.arguments_info
@@ -212,8 +215,8 @@ class ExpectedStruct(ctypes.Structure):
 
 @pytest.mark.parametrize("nargs", [0, 1, 2, 3, 16])
 @pytest.mark.parametrize("c_type_name,c_type", [("int", ctypes.c_int), ("short", ctypes.c_short)], ids=["int", "short"])
-def test_num_arguments(init_cuda, nargs, c_type_name, c_type, cuda12_prerequisite_check):
-    if not cuda12_prerequisite_check:
+def test_num_arguments(init_cuda, nargs, c_type_name, c_type, cuda12_4_prerequisite_check):
+    if not cuda12_4_prerequisite_check:
         pytest.skip("Test requires CUDA 12")
     args_str = ", ".join([f"{c_type_name} p_{i}" for i in range(nargs)])
     src = f"__global__ void foo{nargs}({args_str}) {{ }}"
@@ -235,8 +238,8 @@ class ExpectedStruct(ctypes.Structure):
     assert all([actual.size == expected.size for actual, expected in zip(arg_info, members)])
 
 
-def test_num_args_error_handling(deinit_all_contexts_function, cuda12_prerequisite_check):
-    if not cuda12_prerequisite_check:
+def test_num_args_error_handling(deinit_all_contexts_function, cuda12_4_prerequisite_check):
+    if not cuda12_4_prerequisite_check:
         pytest.skip("Test requires CUDA 12")
     src = "__global__ void foo(int a) { }"
     prog = Program(src, code_type="c++")