mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
a5cc151b8c
Summary: Fixes https://github.com/pytorch/pytorch/issues/43709 Pull Request resolved: https://github.com/pytorch/pytorch/pull/44747 Reviewed By: ezyang Differential Revision: D23717927 Pulled By: malfet fbshipit-source-id: c46fbcf5a55895cb984dd4c5301fbcb784fc17d5
54 lines
1.8 KiB
CMake
54 lines
1.8 KiB
CMake
if(__EIGEN_BLAS_INCLUDED)
|
|
return()
|
|
endif()
|
|
set(__EIGEN_BLAS_INCLUDED TRUE)
|
|
|
|
if(NOT INTERN_BUILD_MOBILE OR NOT INTERN_USE_EIGEN_BLAS)
|
|
return()
|
|
endif()
|
|
|
|
##############################################################################
|
|
# Eigen BLAS is built together with Libtorch mobile.
|
|
# By default, it builds code from third-party/eigen/blas submodule.
|
|
##############################################################################
|
|
|
|
set(CAFFE2_THIRD_PARTY_ROOT ${PROJECT_SOURCE_DIR}/third_party)
|
|
set(EIGEN_BLAS_SRC_DIR "${CAFFE2_THIRD_PARTY_ROOT}/eigen/blas" CACHE STRING "Eigen BLAS source directory")
|
|
|
|
set(EigenBlas_SRCS
|
|
${EIGEN_BLAS_SRC_DIR}/single.cpp
|
|
${EIGEN_BLAS_SRC_DIR}/double.cpp
|
|
${EIGEN_BLAS_SRC_DIR}/complex_single.cpp
|
|
${EIGEN_BLAS_SRC_DIR}/complex_double.cpp
|
|
${EIGEN_BLAS_SRC_DIR}/xerbla.cpp
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/srotm.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/srotmg.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/drotm.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/drotmg.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/lsame.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/dspmv.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/ssbmv.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/chbmv.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/sspmv.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/zhbmv.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/chpmv.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/dsbmv.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/zhpmv.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/dtbmv.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/stbmv.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/ctbmv.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/ztbmv.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/d_cnjg.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/r_cnjg.c
|
|
${EIGEN_BLAS_SRC_DIR}/f2c/complexdots.c
|
|
)
|
|
|
|
add_library(eigen_blas STATIC ${EigenBlas_SRCS})
|
|
|
|
# We build static versions of eigen blas but link into a shared library, so they need PIC.
|
|
set_property(TARGET eigen_blas PROPERTY POSITION_INDEPENDENT_CODE ON)
|
|
|
|
install(TARGETS eigen_blas
|
|
LIBRARY DESTINATION lib
|
|
ARCHIVE DESTINATION lib)
|