Restore Runtime.is_supported_version()

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/device_init.py

@@ -133,7 +133,7 @@ def is_supported_version():
     - Generating an error or otherwise preventing the use of CUDA.
     """
 
-    return runtime.get_version()[0] in (12, 13)
+    return runtime.is_supported_version()
 
 
 def cuda_error():