diff --git a/cupy_backends/cuda/cupy_cusparse.h b/cupy_backends/cuda/cupy_cusparse.h index 9eb3a0429f9..863c5feb9ee 100644 --- a/cupy_backends/cuda/cupy_cusparse.h +++ b/cupy_backends/cuda/cupy_cusparse.h @@ -599,7 +599,7 @@ cusparseStatus_t cusparseSpGEMM_copy(...) { #endif // #if CUSPARSE_VERSION < 11100 #if CUSPARSE_VERSION < 11300 -// Types and macros added in cuSparse 11.3 (CUDA 11.2) +// Types and macros added in cuSparse 11.3 (CUDA 11.1.1) typedef enum {} cusparseSparseToDenseAlg_t; typedef enum {} cusparseDenseToSparseAlg_t; diff --git a/cupy_backends/cuda/libs/cusparse.pyx b/cupy_backends/cuda/libs/cusparse.pyx index 1d5ab66331e..4e866fcc1ba 100644 --- a/cupy_backends/cuda/libs/cusparse.pyx +++ b/cupy_backends/cuda/libs/cusparse.pyx @@ -1424,8 +1424,13 @@ cdef Status cusparseDenseToSparse_convert( cdef load_functions(libname, prefix): lib = SoftLink(libname, prefix) + + # cuSPARSE 11.3.1+ (CUDA 11.2.0+) global _cusparseCreateCsc _cusparseCreateCsc = lib.get_func('CreateCsc') + + # cuSPARSE 11.3+ (CUDA 11.1.1+) + # Note: CUDA 11.1.0 contains cuSPARSE 11.2.0.275 global _cusparseSparseToDense_bufferSize _cusparseSparseToDense_bufferSize = lib.get_func('SparseToDense_bufferSize') # NOQA global _cusparseSparseToDense @@ -1437,7 +1442,7 @@ cdef load_functions(libname, prefix): global _cusparseDenseToSparse_convert _cusparseDenseToSparse_convert = lib.get_func('DenseToSparse_convert') -IF 11020 <= CUPY_CUDA_VERSION < 12000: +IF 11010 <= CUPY_CUDA_VERSION < 12000: if _sys.platform == 'linux': _libname = 'libcusparse.so.11' else: