Add support for CUDA 11 (#300).

Author: lebedov

docs/source/changes.rst

@@ -6,6 +6,7 @@ Change Log
 Release 0.5.4 (Under Development)
 ---------------------------------
 * Add wrappers for matrix-matrix multiplication of batches of matrices (#278).
+* Add support for CUDA 11.*.
 
 Release 0.5.3 (May 26, 2019)
 ----------------------------

skcuda/cublas.py

@@ -23,8 +23,8 @@
 from . import utils
 
 # Load library:
-_linux_version_list = [10.2, 10.1, 10.0, 9.2, 9.1, 9.0, 8.0, 7.5, 7.0, 6.5, 6.0, 5.5, 5.0, 4.0]
-_win32_version_list = [10, 10, 100, 92, 91, 90, 80, 75, 70, 65, 60, 55, 50, 40]
+_linux_version_list = [11.0, 10.2, 10.1, 10.0, 9.2, 9.1, 9.0, 8.0, 7.5, 7.0, 6.5, 6.0, 5.5, 5.0, 4.0]
+_win32_version_list = [11, 10, 10, 100, 92, 91, 90, 80, 75, 70, 65, 60, 55, 50, 40]
 if 'linux' in sys.platform:
     _libcublas_libname_list = ['libcublas.so'] + \
                               ['libcublas.so.%s' % v for v in _linux_version_list]

skcuda/cudart.py

@@ -8,8 +8,8 @@
 import numpy as np
 
 # Load library:
-_linux_version_list = [10.2, 10.1, 10.0, 9.2, 9.1, 9.0, 8.0, 7.5, 7.0, 6.5, 6.0, 5.5, 5.0, 4.0]
-_win32_version_list = [102, 101, 100, 92, 91, 90, 80, 75, 70, 65, 60, 55, 50, 40]
+_linux_version_list = [11.0, 10.2, 10.1, 10.0, 9.2, 9.1, 9.0, 8.0, 7.5, 7.0, 6.5, 6.0, 5.5, 5.0, 4.0]
+_win32_version_list = [110, 102, 101, 100, 92, 91, 90, 80, 75, 70, 65, 60, 55, 50, 40]
 if 'linux' in sys.platform:
     _libcudart_libname_list = ['libcudart.so'] + \
                               ['libcudart.so.%s' % v for v in _linux_version_list]

skcuda/cufft.py

@@ -12,8 +12,8 @@
 import sys
 
 # Load library:
-_linux_version_list = [10.1, 10.0, 9.2, 9.1, 9.0, 8.0, 7.5, 7.0, 6.5, 6.0, 5.5, 5.0, 4.0]
-_win32_version_list = [10, 10, 100, 92, 91, 90, 80, 75, 70, 65, 60, 55, 50, 40]
+_linux_version_list = [11.0, 10.1, 10.0, 9.2, 9.1, 9.0, 8.0, 7.5, 7.0, 6.5, 6.0, 5.5, 5.0, 4.0]
+_win32_version_list = [110, 10, 10, 100, 92, 91, 90, 80, 75, 70, 65, 60, 55, 50, 40]
 if 'linux' in sys.platform:
     _libcufft_libname_list = ['libcufft.so'] + \
                              ['libcufft.so.%s' % v for v in _linux_version_list]

skcuda/cusolver.py

@@ -22,8 +22,8 @@
 from . import cublas
 
 # Load library:
-_linux_version_list = [10.2, 10.1, 10.0, 9.2, 9.1, 9.0, 8.0, 7.5, 7.0]
-_win32_version_list = [10, 10, 100, 92, 91, 90, 80, 75, 70]
+_linux_version_list = [11.0, 10.2, 10.1, 10.0, 9.2, 9.1, 9.0, 8.0, 7.5, 7.0]
+_win32_version_list = [110, 10, 10, 100, 92, 91, 90, 80, 75, 70]
 if 'linux' in sys.platform:
     _libcusolver_libname_list = ['libcusolver.so'] + \
                                 ['libcusolver.so.%s' % v for v in _linux_version_list]

skcuda/cusparse.py

@@ -18,8 +18,8 @@
 import cuda
 
 # Load library:
-_version_list = [10.1, 10.0, 9.2, 9.1, 9.0, 8.0, 7.5, 7.0, 6.5, 6.0, 5.5, 5.0, 4.0]
-_win32_version_list = [10, 10, 100, 92, 91, 90, 80, 75, 70, 65, 60, 55, 50, 40]
+_version_list = [11.0, 10.1, 10.0, 9.2, 9.1, 9.0, 8.0, 7.5, 7.0, 6.5, 6.0, 5.5, 5.0, 4.0]
+_win32_version_list = [110, 10, 10, 100, 92, 91, 90, 80, 75, 70, 65, 60, 55, 50, 40]
 if 'linux' in sys.platform:
     _libcusparse_libname_list = ['libcusparse.so'] + \
                                 ['libcusparse.so.%s' % v for v in _linux_version_list]