Skip to content

Commit 8a62e37

Browse files
authored
support (#68901)
1 parent 63b8765 commit 8a62e37

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

paddle/phi/backends/dynload/dynamic_loader.cc

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)