File tree Expand file tree Collapse file tree 1 file changed +21
-0
lines changed
paddle/phi/backends/dynload Expand file tree Collapse file tree 1 file changed +21
-0
lines changed Original file line number Diff line number Diff line change @@ -417,6 +417,27 @@ void* GetCublasLtDsoHandle() {
417417 " temporarily no longer supports" );
418418 return nullptr ;
419419 }
420+ #elif defined(_WIN32) && defined(PADDLE_WITH_CUDA)
421+ if (CUDA_VERSION >= 11000 && CUDA_VERSION < 12000 ) {
422+ #ifdef PADDLE_WITH_PIP_CUDA_LIBRARIES
423+ return GetDsoHandleFromSearchPath (FLAGS_cuda_dir, " cublasLt64_11.dll" );
424+ #else
425+ return GetDsoHandleFromSearchPath (
426+ FLAGS_cuda_dir, win_cublas_lib, true , {cuda_lib_path});
427+ #endif
428+ } else if (CUDA_VERSION >= 12000 && CUDA_VERSION < 13000 ) {
429+ #ifdef PADDLE_WITH_PIP_CUDA_LIBRARIES
430+ return GetDsoHandleFromSearchPath (FLAGS_cuda_dir, " cublasLt64_12.dll" );
431+ #else
432+ return GetDsoHandleFromSearchPath (
433+ FLAGS_cuda_dir, win_cublas_lib, true , {cuda_lib_path});
434+ #endif
435+ } else {
436+ std::string warning_msg (
437+ " Your CUDA_VERSION is less than 11 or greater than 12, paddle "
438+ " temporarily no longer supports" );
439+ return nullptr ;
440+ }
420441#elif !defined(__linux__) && defined(PADDLE_WITH_CUDA) && CUDA_VERSION >= 10010
421442 return GetDsoHandleFromSearchPath (FLAGS_cuda_dir, " libcublasLt.so" );
422443#elif defined(PADDLE_WITH_HIP)
You can’t perform that action at this time.
0 commit comments