Fix the cuda.is_supported_version() API (#571)

Author: gmarkall

numba_cuda/numba/cuda/cudadrv/runtime.py

@@ -10,11 +10,19 @@
 
 from numba.cuda.cudadrv.nvrtc import _get_nvrtc_version
 
+SUPPORTED_TOOLKIT_MAJOR_VERSIONS = (12, 13)
+
 
 class Runtime:
     def get_version(self):
         return _get_nvrtc_version()
 
+    def is_supported_version(self):
+        """
+        Returns True if the CUDA Runtime is a supported version.
+        """
+        return self.get_version()[0] in SUPPORTED_TOOLKIT_MAJOR_VERSIONS
+
 
 runtime = Runtime()
 

numba_cuda/numba/cuda/simulator/api.py

@@ -160,3 +160,7 @@ def jitwrapper(fn):
 def defer_cleanup():
     # No effect for simulator
     yield
+
+
+def is_supported_version():
+    return True

numba_cuda/numba/cuda/tests/cudadrv/test_detect.py

@@ -25,6 +25,16 @@ def test_cuda_detect(self):
         self.assertIn("CUDA devices", output)
 
 
+class TestSupportedVersion(CUDATestCase):
+    def test_is_supported_version(self):
+        # Exercise the `cuda.is_supported_version()` API.
+        #
+        # Assume for the purpose of the test that we're running on a supported
+        # toolkit version; if not, there's not much point in running the test
+        # suite.
+        self.assertTrue(cuda.is_supported_version())
+
+
 @skip_under_cuda_memcheck("Hangs cuda-memcheck")
 class TestCUDAFindLibs(CUDATestCase):
     def run_cmd(self, cmdline, env):