diff options
author | Igor S. Gerasimov <i.s.ger@ya.ru> | 2021-08-07 08:13:33 (GMT) |
---|---|---|
committer | Brad King <brad.king@kitware.com> | 2021-09-22 19:16:22 (GMT) |
commit | c4d4f0708b9c8aa0cba6d8d031aee7d30798dba6 (patch) | |
tree | 30ab00c96b727cba8a5f9e24d40129ffd4b20da6 /Modules/FindBLAS.cmake | |
parent | 2531e464dfefaacdd06c6b28e7f59897510a52c1 (diff) | |
download | CMake-c4d4f0708b9c8aa0cba6d8d031aee7d30798dba6.zip CMake-c4d4f0708b9c8aa0cba6d8d031aee7d30798dba6.tar.gz CMake-c4d4f0708b9c8aa0cba6d8d031aee7d30798dba6.tar.bz2 |
FindBLAS: Move NVHPC from Generic BLAS
Diffstat (limited to 'Modules/FindBLAS.cmake')
-rw-r--r-- | Modules/FindBLAS.cmake | 21 |
1 files changed, 20 insertions, 1 deletions
diff --git a/Modules/FindBLAS.cmake b/Modules/FindBLAS.cmake index a44af4d..5743f9d 100644 --- a/Modules/FindBLAS.cmake +++ b/Modules/FindBLAS.cmake @@ -1153,9 +1153,28 @@ if(NOT BLAS_LIBRARIES unset(_ssl2_suffix) endif() +# BLAS in nVidia HPC SDK? (https://developer.nvidia.com/hpc-sdk) +if(BLA_VENDOR STREQUAL "NVHPC" OR BLA_VENDOR STREQUAL "All") + set(_blas_nvhpc_lib "blas") + + if(NOT BLAS_LIBRARIES) + check_blas_libraries( + BLAS_LIBRARIES + BLAS + sgemm + "" + "${_blas_nvhpc_lib}" + "" + "" + "" + ) + endif() + + unset(_blas_nvhpc_lib) +endif() + # Generic BLAS library? if(BLA_VENDOR STREQUAL "Generic" OR - BLA_VENDOR STREQUAL "NVHPC" OR BLA_VENDOR STREQUAL "All") if(NOT BLAS_LIBRARIES) check_blas_libraries( |