pytorch

1

2

3
SET(Flexi_BLAS_INCLUDE_SEARCH_PATHS
4
  /usr/include
5
  /usr/include/flexiblas
6
  /usr/include/flexiblas-base
7
  /usr/local/include
8
  /usr/local/include/flexiblas
9
  /usr/local/include/flexiblas-base
10
  /usr/local/opt/flexiblas/include
11
  /opt/Flexiblas/include
12
  $ENV{FlexiBLAS_HOME}
13
  $ENV{FlexiBLAS_HOME}/include
14
)
15

16
SET(Flexi_BLAS_LIB_SEARCH_PATHS
17
        /lib/
18
        /lib/flexiblas-base
19
        /lib64/
20
        /usr/lib
21
        /usr/lib/flexiblas-base
22
        /usr/lib64
23
        /usr/local/lib
24
        /usr/local/lib64
25
        /usr/local/opt/flexiblas/lib
26
        /opt/FlexiBLAS/lib
27
        $ENV{FlexiBLAS}
28
        $ENV{FlexiBLAS}/lib
29
        $ENV{FlexiBLAS_HOME}
30
        $ENV{FlexiBLAS_HOME}/lib
31
 )
32

33
FIND_PATH(FlexiBLAS_INCLUDE_DIR NAMES cblas.h PATHS ${Flexi_BLAS_INCLUDE_SEARCH_PATHS})
34
FIND_LIBRARY(FlexiBLAS_LIB NAMES flexiblas PATHS ${Flexi_BLAS_LIB_SEARCH_PATHS})
35

36
SET(FlexiBLAS_FOUND ON)
37

38
#    Check include files
39
IF(NOT FlexiBLAS_INCLUDE_DIR)
40
    SET(FlexiBLAS_FOUND OFF)
41
    MESSAGE(STATUS "Could not find FlexiBLAS include. Turning FlexiBLAS_FOUND off")
42
ENDIF()
43

44
#    Check libraries
45
IF(NOT FlexiBLAS_LIB)
46
    SET(FlexiBLAS_FOUND OFF)
47
    MESSAGE(STATUS "Could not find FlexiBLAS lib. Turning FlexiBLAS_FOUND off")
48
ENDIF()
49

50
IF (FlexiBLAS_FOUND)
51
  IF (NOT FlexiBLAS_FIND_QUIETLY)
52
    MESSAGE(STATUS "Found FlexiBLAS libraries: ${FlexiBLAS_LIB}")
53
    MESSAGE(STATUS "Found FlexiBLAS include: ${FlexiBLAS_INCLUDE_DIR}")
54
  ENDIF (NOT FlexiBLAS_FIND_QUIETLY)
55
ELSE (FlexiBLAS_FOUND)
56
  IF (FlexiBLAS_FIND_REQUIRED)
57
    MESSAGE(FATAL_ERROR "Could not find FlexiBLAS")
58
  ENDIF (FlexiBLAS_FIND_REQUIRED)
59
ENDIF (FlexiBLAS_FOUND)
60

61
MARK_AS_ADVANCED(
62
    FlexiBLAS_INCLUDE_DIR
63
    FlexiBLAS_LIB
64
    FlexiBLAS
65
)
66