frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixes detection of ArmPL on Linux platform (#150031)

Browse Source 
On Linux it failed to detect that there is bin directory as it wasn't looking for armpl-info which is the only file that is in that directory on Linux and also adding link to math library as it is required to link against when checking for LAPACK functions.

Fixes #149610

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150031
Approved by: https://github.com/fadara01, https://github.com/malfet
This commit is contained in:
Milos Puzovic
2025-05-07 19:47:16 +00:00
committed by PyTorch MergeBot
parent f5f8f637a5
commit 642e9305eb
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cmake/Modules/FindAPL.cmake
View File

@ -29,7 +29,7 @@ IF(NOT APL_LIB_DIR)
ENDIF()
# Check bin file
FIND_PATH(APL_BIN_DIR NAMES armpl_lp64.dll libarmpl_lp64.a PATHS ${APL_BIN_SEARCH_PATHS})
FIND_PATH(APL_BIN_DIR NAMES armpl_lp64.dll armpl-info PATHS ${APL_BIN_SEARCH_PATHS})
IF(NOT APL_BIN_DIR)
SET(APL_FOUND OFF)
MESSAGE(STATUS "Could not verify APL bin directory. Turning APL_FOUND off")

3
cmake/Modules/FindLAPACK.cmake
View File

@ -226,6 +226,9 @@ if(BLAS_FOUND)
#Arm Performance Libraries
IF((NOT LAPACK_INFO) AND (BLAS_INFO STREQUAL "apl"))
SET(CMAKE_REQUIRED_LIBRARIES ${BLAS_LIBRARIES})
if(UNIX)
list(APPEND CMAKE_REQUIRED_LIBRARIES -lm)
endif(UNIX)
check_function_exists("cheev_" APL_LAPACK_WORKS)
if(APL_LAPACK_WORKS)
check_function_exists("cgesdd_" LAPACK_CGESDD_WORKS)