diff options
| author | Robert Maynard <rmaynard@nvidia.com> | 2021-05-05 13:33:39 (GMT) |
|---|---|---|
| committer | Robert Maynard <rmaynard@nvidia.com> | 2021-05-05 16:04:32 (GMT) |
| commit | 2c9e623e3194519da0714c77cba48592d6658cda (patch) | |
| tree | 162c62ecbdb0bcd52ed262a1df9c393e4e684ada /Modules | |
| parent | de0c8b4d98d71280abd585e17f6bd7b20d4de959 (diff) | |
| download | CMake-2c9e623e3194519da0714c77cba48592d6658cda.zip CMake-2c9e623e3194519da0714c77cba48592d6658cda.tar.gz CMake-2c9e623e3194519da0714c77cba48592d6658cda.tar.bz2 | |
Find{BLAS,LAPACK}: Add support for the NVHPC LAPACK library
Diffstat (limited to 'Modules')
| -rw-r--r-- | Modules/FindBLAS.cmake | 8 | ||||
| -rw-r--r-- | Modules/FindLAPACK.cmake | 20 |
2 files changed, 27 insertions, 1 deletions
diff --git a/Modules/FindBLAS.cmake b/Modules/FindBLAS.cmake index 024ff14..0f8360a 100644 --- a/Modules/FindBLAS.cmake +++ b/Modules/FindBLAS.cmake @@ -57,6 +57,7 @@ The following variables may be set to influence this module's behavior: * ``EML_mt`` * ``Fujitsu_SSL2`` (Fujitsu serial blas / lapack) * ``Fujitsu_SSL2BLAMP`` (Fujitsu parallel blas / lapack) + * ``NVHPC`` * ``Generic`` .. versionadded:: 3.6 @@ -82,6 +83,9 @@ The following variables may be set to influence this module's behavior: Elbrus Math Library support (``EML``, ``EML_mt``). Fujitsu SSL2 Library support (``Fujitsu_SSL2``, ``Fujitsu_SSL2BLAMP``) + .. versionadded:: 3.21 + NVHPC support + ``BLA_F95`` if ``ON`` tries to find the BLAS95 interfaces @@ -1062,7 +1066,9 @@ if(NOT BLAS_LIBRARIES AND endif() # Generic BLAS library? -if(BLA_VENDOR STREQUAL "Generic" OR BLA_VENDOR STREQUAL "All") +if(BLA_VENDOR STREQUAL "Generic" OR + BLA_VENDOR STREQUAL "NVHPC" OR + BLA_VENDOR STREQUAL "All") if(NOT BLAS_LIBRARIES) check_blas_libraries( BLAS_LIBRARIES diff --git a/Modules/FindLAPACK.cmake b/Modules/FindLAPACK.cmake index ab6214b..fa2300d 100644 --- a/Modules/FindLAPACK.cmake +++ b/Modules/FindLAPACK.cmake @@ -47,6 +47,7 @@ The following variables may be set to influence this module's behavior: * ``EML_mt`` * ``Fujitsu_SSL2`` (Fujitsu serial blas / lapack) * ``Fujitsu_SSL2BLAMP`` (Fujitsu parallel blas / lapack) + * ``NVHPC`` * ``Generic`` .. versionadded:: 3.6 @@ -72,6 +73,9 @@ The following variables may be set to influence this module's behavior: Elbrus Math Library support (``EML``, ``EML_mt``). Fujitsu SSL2 Library support (``Fujitsu_SSL2``, ``Fujitsu_SSL2BLAMP``) + .. versionadded:: 3.21 + NVHPC support + ``BLA_F95`` if ``ON`` tries to find the BLAS95/LAPACK95 interfaces @@ -615,6 +619,22 @@ if(NOT LAPACK_NOT_FOUND_MESSAGE) unset(_ssl2_suffix) endif() + # NVHPC Library? + if(NOT LAPACK_LIBRARIES + AND BLA_VENDOR MATCHES "NVHPC" OR BLA_VENDOR STREQUAL "All") + check_lapack_libraries( + LAPACK_LIBRARIES + LAPACK + cheev + "" + "lapack" + "-fortranlibs" + "" + "" + "${BLAS_LIBRARIES}" + ) + endif() + # Generic LAPACK library? if(NOT LAPACK_LIBRARIES AND (BLA_VENDOR STREQUAL "Generic" |