mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
007fe949aa
Summary:
While trying to build PyTorch with BLIS as the backend library,
we found a build issue due to some missing include files.
This was caused by a missing directory in the search path.
This patch adds that path in FindBLIS.cmake.
Fixes #{issue number}
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58166
Reviewed By: zou3519
Differential Revision: D28640460
Pulled By: malfet
fbshipit-source-id: d0cd3a680718a0a45788c46a502871b88fbadd52
71 lines
1.7 KiB
CMake
71 lines
1.7 KiB
CMake
# - 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
|
|
)
|