xref: /aosp_15_r20/external/pytorch/cmake/Modules/FindOpenBLAS.cmake (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1
2
3SET(Open_BLAS_INCLUDE_SEARCH_PATHS
4  /usr/include
5  /usr/include/openblas
6  /usr/include/openblas-base
7  /usr/local/include
8  /usr/local/include/openblas
9  /usr/local/include/openblas-base
10  /usr/local/opt/openblas/include
11  /opt/OpenBLAS/include
12  $ENV{OpenBLAS_HOME}
13  $ENV{OpenBLAS_HOME}/include
14  $ENV{OpenBLAS_HOME}/include/openblas
15)
16
17SET(Open_BLAS_LIB_SEARCH_PATHS
18        /lib/
19        /lib/openblas-base
20        /lib64/
21        /usr/lib
22        /usr/lib/openblas-base
23        /usr/lib64
24        /usr/local/lib
25        /usr/local/lib64
26        /usr/local/opt/openblas/lib
27        /opt/OpenBLAS/lib
28        $ENV{OpenBLAS}
29        $ENV{OpenBLAS}/lib
30        $ENV{OpenBLAS_HOME}
31        $ENV{OpenBLAS_HOME}/lib
32 )
33
34FIND_PATH(OpenBLAS_INCLUDE_DIR NAMES cblas.h PATHS ${Open_BLAS_INCLUDE_SEARCH_PATHS})
35FIND_LIBRARY(OpenBLAS_LIB NAMES openblas PATHS ${Open_BLAS_LIB_SEARCH_PATHS})
36
37SET(OpenBLAS_FOUND ON)
38
39#    Check include files
40IF(NOT OpenBLAS_INCLUDE_DIR)
41    SET(OpenBLAS_FOUND OFF)
42    MESSAGE(STATUS "Could not find OpenBLAS include. Turning OpenBLAS_FOUND off")
43ENDIF()
44
45#    Check libraries
46IF(NOT OpenBLAS_LIB)
47    SET(OpenBLAS_FOUND OFF)
48    MESSAGE(STATUS "Could not find OpenBLAS lib. Turning OpenBLAS_FOUND off")
49ENDIF()
50
51IF (OpenBLAS_FOUND)
52  IF (NOT OpenBLAS_FIND_QUIETLY)
53    MESSAGE(STATUS "Found OpenBLAS libraries: ${OpenBLAS_LIB}")
54    MESSAGE(STATUS "Found OpenBLAS include: ${OpenBLAS_INCLUDE_DIR}")
55  ENDIF (NOT OpenBLAS_FIND_QUIETLY)
56ELSE (OpenBLAS_FOUND)
57  IF (OpenBLAS_FIND_REQUIRED)
58    MESSAGE(FATAL_ERROR "Could not find OpenBLAS")
59  ENDIF (OpenBLAS_FIND_REQUIRED)
60ENDIF (OpenBLAS_FOUND)
61
62MARK_AS_ADVANCED(
63    OpenBLAS_INCLUDE_DIR
64    OpenBLAS_LIB
65    OpenBLAS
66)
67