xref: /aosp_15_r20/external/pytorch/cmake/Modules/FindFlexiBLAS.cmake (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker
2*da0073e9SAndroid Build Coastguard Worker
3*da0073e9SAndroid Build Coastguard WorkerSET(Flexi_BLAS_INCLUDE_SEARCH_PATHS
4*da0073e9SAndroid Build Coastguard Worker  /usr/include
5*da0073e9SAndroid Build Coastguard Worker  /usr/include/flexiblas
6*da0073e9SAndroid Build Coastguard Worker  /usr/include/flexiblas-base
7*da0073e9SAndroid Build Coastguard Worker  /usr/local/include
8*da0073e9SAndroid Build Coastguard Worker  /usr/local/include/flexiblas
9*da0073e9SAndroid Build Coastguard Worker  /usr/local/include/flexiblas-base
10*da0073e9SAndroid Build Coastguard Worker  /usr/local/opt/flexiblas/include
11*da0073e9SAndroid Build Coastguard Worker  /opt/Flexiblas/include
12*da0073e9SAndroid Build Coastguard Worker  $ENV{FlexiBLAS_HOME}
13*da0073e9SAndroid Build Coastguard Worker  $ENV{FlexiBLAS_HOME}/include
14*da0073e9SAndroid Build Coastguard Worker)
15*da0073e9SAndroid Build Coastguard Worker
16*da0073e9SAndroid Build Coastguard WorkerSET(Flexi_BLAS_LIB_SEARCH_PATHS
17*da0073e9SAndroid Build Coastguard Worker        /lib/
18*da0073e9SAndroid Build Coastguard Worker        /lib/flexiblas-base
19*da0073e9SAndroid Build Coastguard Worker        /lib64/
20*da0073e9SAndroid Build Coastguard Worker        /usr/lib
21*da0073e9SAndroid Build Coastguard Worker        /usr/lib/flexiblas-base
22*da0073e9SAndroid Build Coastguard Worker        /usr/lib64
23*da0073e9SAndroid Build Coastguard Worker        /usr/local/lib
24*da0073e9SAndroid Build Coastguard Worker        /usr/local/lib64
25*da0073e9SAndroid Build Coastguard Worker        /usr/local/opt/flexiblas/lib
26*da0073e9SAndroid Build Coastguard Worker        /opt/FlexiBLAS/lib
27*da0073e9SAndroid Build Coastguard Worker        $ENV{FlexiBLAS}
28*da0073e9SAndroid Build Coastguard Worker        $ENV{FlexiBLAS}/lib
29*da0073e9SAndroid Build Coastguard Worker        $ENV{FlexiBLAS_HOME}
30*da0073e9SAndroid Build Coastguard Worker        $ENV{FlexiBLAS_HOME}/lib
31*da0073e9SAndroid Build Coastguard Worker )
32*da0073e9SAndroid Build Coastguard Worker
33*da0073e9SAndroid Build Coastguard WorkerFIND_PATH(FlexiBLAS_INCLUDE_DIR NAMES cblas.h PATHS ${Flexi_BLAS_INCLUDE_SEARCH_PATHS})
34*da0073e9SAndroid Build Coastguard WorkerFIND_LIBRARY(FlexiBLAS_LIB NAMES flexiblas PATHS ${Flexi_BLAS_LIB_SEARCH_PATHS})
35*da0073e9SAndroid Build Coastguard Worker
36*da0073e9SAndroid Build Coastguard WorkerSET(FlexiBLAS_FOUND ON)
37*da0073e9SAndroid Build Coastguard Worker
38*da0073e9SAndroid Build Coastguard Worker#    Check include files
39*da0073e9SAndroid Build Coastguard WorkerIF(NOT FlexiBLAS_INCLUDE_DIR)
40*da0073e9SAndroid Build Coastguard Worker    SET(FlexiBLAS_FOUND OFF)
41*da0073e9SAndroid Build Coastguard Worker    MESSAGE(STATUS "Could not find FlexiBLAS include. Turning FlexiBLAS_FOUND off")
42*da0073e9SAndroid Build Coastguard WorkerENDIF()
43*da0073e9SAndroid Build Coastguard Worker
44*da0073e9SAndroid Build Coastguard Worker#    Check libraries
45*da0073e9SAndroid Build Coastguard WorkerIF(NOT FlexiBLAS_LIB)
46*da0073e9SAndroid Build Coastguard Worker    SET(FlexiBLAS_FOUND OFF)
47*da0073e9SAndroid Build Coastguard Worker    MESSAGE(STATUS "Could not find FlexiBLAS lib. Turning FlexiBLAS_FOUND off")
48*da0073e9SAndroid Build Coastguard WorkerENDIF()
49*da0073e9SAndroid Build Coastguard Worker
50*da0073e9SAndroid Build Coastguard WorkerIF (FlexiBLAS_FOUND)
51*da0073e9SAndroid Build Coastguard Worker  IF (NOT FlexiBLAS_FIND_QUIETLY)
52*da0073e9SAndroid Build Coastguard Worker    MESSAGE(STATUS "Found FlexiBLAS libraries: ${FlexiBLAS_LIB}")
53*da0073e9SAndroid Build Coastguard Worker    MESSAGE(STATUS "Found FlexiBLAS include: ${FlexiBLAS_INCLUDE_DIR}")
54*da0073e9SAndroid Build Coastguard Worker  ENDIF (NOT FlexiBLAS_FIND_QUIETLY)
55*da0073e9SAndroid Build Coastguard WorkerELSE (FlexiBLAS_FOUND)
56*da0073e9SAndroid Build Coastguard Worker  IF (FlexiBLAS_FIND_REQUIRED)
57*da0073e9SAndroid Build Coastguard Worker    MESSAGE(FATAL_ERROR "Could not find FlexiBLAS")
58*da0073e9SAndroid Build Coastguard Worker  ENDIF (FlexiBLAS_FIND_REQUIRED)
59*da0073e9SAndroid Build Coastguard WorkerENDIF (FlexiBLAS_FOUND)
60*da0073e9SAndroid Build Coastguard Worker
61*da0073e9SAndroid Build Coastguard WorkerMARK_AS_ADVANCED(
62*da0073e9SAndroid Build Coastguard Worker    FlexiBLAS_INCLUDE_DIR
63*da0073e9SAndroid Build Coastguard Worker    FlexiBLAS_LIB
64*da0073e9SAndroid Build Coastguard Worker    FlexiBLAS
65*da0073e9SAndroid Build Coastguard Worker)
66