xref: /aosp_15_r20/external/pytorch/cmake/Modules/FindBLIS.cmake (revision da0073e96a02ea20f0ac840b70461e3646d07c45)

# - Find BLIS library
#
# This module sets the following variables:
#  BLIS_FOUND - set to true if a library implementing CBLAS interface is found.
#  BLIS_INCLUDE_DIR - path to include dir.
#  BLIS_LIB - list of libraries for BLIS.
#
# CPU only Dockerfile to build with AMD BLIS is available at the location
# pytorch/docker/pytorch/cpu-blis/Dockerfile
#


SET(BLIS_INCLUDE_SEARCH_PATHS
  /usr/include/blis
  /usr/local/include
  /usr/local/include/blis
  /opt/blis/include
  $ENV{BLIS_HOME}
  $ENV{BLIS_HOME}/include
  $ENV{BLIS_HOME}/include/blis
)

SET(BLIS_LIB_SEARCH_PATHS
  /lib/blis
  /lib64/blis
  /usr/lib/blis
  /usr/lib64/blis
  /usr/local/blis/lib
  /opt/blis/lib
  $ENV{BLIS_HOME}
  $ENV{BLIS_HOME}/lib
)

FIND_PATH(BLIS_INCLUDE_DIR NAMES cblas.h blis.h
          PATHS ${BLIS_INCLUDE_SEARCH_PATHS})
#    Check include files
IF(NOT BLIS_INCLUDE_DIR)
        SET(BLIS_FOUND OFF)
        MESSAGE(WARNING "Could not find BLIS include. Turning BLIS_FOUND off")
        RETURN()
ENDIF()


FIND_LIBRARY(BLIS_LIB NAMES blis PATHS ${BLIS_LIB_SEARCH_PATHS})
#    Check libraries
IF(NOT BLIS_LIB)
        SET(BLIS_FOUND OFF)
        MESSAGE(WARNING "Could not find BLIS lib. Turning BLIS_FOUND off")
        RETURN()
ENDIF()

SET(BLIS_FOUND ON)

IF(BLIS_FOUND)
        IF(NOT BLIS_FIND_QUIETLY)
                MESSAGE(STATUS "Found BLIS libraries: ${BLIS_LIB}")
                MESSAGE(STATUS "Found BLIS include: ${BLIS_INCLUDE_DIR}")
        ENDIF()
ELSE()
        MESSAGE(FATAL_ERROR "Could not find BLIS")
ENDIF()

INCLUDE(FindPackageHandleStandardArgs)
FIND_PACKAGE_HANDLE_STANDARD_ARGS(BLIS DEFAULT_MSG BLIS_INCLUDE_DIR BLIS_LIB)

MARK_AS_ADVANCED(
        BLIS_INCLUDE_DIR
        BLIS_LIB
        blis
)