forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathEigenBLAS.cmake
53 lines (46 loc) · 1.81 KB
/
EigenBLAS.cmake
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
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)