File tree Expand file tree Collapse file tree 2 files changed +12
-9
lines changed Expand file tree Collapse file tree 2 files changed +12
-9
lines changed Original file line number Diff line number Diff line change @@ -65,6 +65,7 @@ option(REPLACE_ENFORCE_GLOG "Replace PADDLE_ENFORCE with glog/CHECK for better d
65
65
option (WITH_ANAKIN "Compile with Anakin library" OFF )
66
66
option (WITH_GRPC "Use grpc as the default rpc framework" ${WITH_DISTRIBUTE} )
67
67
option (WITH_BRPC_RDMA "Use brpc rdma as the rpc protocal" OFF )
68
+ option (WITH_SYSTEM_BLAS "Use system blas library" OFF )
68
69
69
70
# CMAKE_BUILD_TYPE
70
71
if (NOT CMAKE_BUILD_TYPE )
Original file line number Diff line number Diff line change @@ -83,18 +83,20 @@ else()
83
83
set (REFERENCE_CBLAS_LIB_SEARCH_PATHS ${REFERENCE_CBLAS_ROOT} /lib )
84
84
endif ()
85
85
86
- find_path (REFERENCE_CBLAS_INCLUDE_DIR NAMES cblas.h PATHS
86
+ if (WITH_SYSTEM_BLAS )
87
+ find_path (REFERENCE_CBLAS_INCLUDE_DIR NAMES cblas.h PATHS
87
88
${REFERENCE_CBLAS_INCLUDE_SEARCH_PATHS} )
88
- find_library (REFERENCE_CBLAS_LIBRARY NAMES cblas PATHS
89
+ find_library (REFERENCE_CBLAS_LIBRARY NAMES cblas PATHS
89
90
${REFERENCE_CBLAS_LIB_SEARCH_PATHS} )
90
91
91
- if (REFERENCE_CBLAS_INCLUDE_DIR AND REFERENCE_CBLAS_LIBRARY )
92
- set (CBLAS_FOUND ON )
93
- set (CBLAS_PROVIDER REFERENCE )
94
- set (CBLAS_INC_DIR ${REFERENCE_CBLAS_INCLUDE_DIR} )
95
- set (CBLAS_LIBRARIES ${REFERENCE_CBLAS_LIBRARY} )
96
- add_definitions (-DPADDLE_USE_REFERENCE_CBLAS )
97
- message (STATUS "Found reference-cblas (include: ${CBLAS_INC_DIR} , library: ${CBLAS_LIBRARIES} )" )
92
+ if (REFERENCE_CBLAS_INCLUDE_DIR AND REFERENCE_CBLAS_LIBRARY )
93
+ set (CBLAS_FOUND ON )
94
+ set (CBLAS_PROVIDER REFERENCE )
95
+ set (CBLAS_INC_DIR ${REFERENCE_CBLAS_INCLUDE_DIR} )
96
+ set (CBLAS_LIBRARIES ${REFERENCE_CBLAS_LIBRARY} )
97
+ add_definitions (-DPADDLE_USE_REFERENCE_CBLAS )
98
+ message (STATUS "Found reference-cblas (include: ${CBLAS_INC_DIR} , library: ${CBLAS_LIBRARIES} )" )
99
+ endif ()
98
100
endif ()
99
101
100
102
if (IOS_USE_VECLIB_FOR_BLAS AND VECLIB_FOUND )
You can’t perform that action at this time.
0 commit comments