File tree Expand file tree Collapse file tree 3 files changed +22
-3
lines changed Expand file tree Collapse file tree 3 files changed +22
-3
lines changed Original file line number Diff line number Diff line change @@ -66,6 +66,8 @@ option(WITH_ARM_FP16 "Use half precision support on armv8.2-a cpu" OFF)
66
66
option (WITH_CONTRIB "Compile the third-party contributation" OFF )
67
67
option (REPLACE_ENFORCE_GLOG "Replace PADDLE_ENFORCE with glog/CHECK for better debug." OFF )
68
68
option (WITH_ANAKIN "Compile with Anakin library" OFF )
69
+ option (ANAKIN_BUILD_FAT_BIN "Build anakin cuda fat-bin lib for all device plantform, ignored when WITH_ANAKIN=OFF" OFF )
70
+ option (ANAKIN_BUILD_CROSS_PLANTFORM "Build anakin lib for any nvidia device plantform. ignored when WITH_ANAKIN=OFF" ON )
69
71
option (WITH_GRPC "Use grpc as the default rpc framework" ${WITH_DISTRIBUTE} )
70
72
option (WITH_BRPC_RDMA "Use brpc rdma as the rpc protocal" OFF )
71
73
option (ON_INFER "Turn on inference optimization." OFF )
Original file line number Diff line number Diff line change @@ -58,19 +58,21 @@ ExternalProject_Add(
58
58
-DPROTOBUF_ROOT=${THIRD_PARTY_PATH}/install/protobuf
59
59
-DMKLML_ROOT=${THIRD_PARTY_PATH}/install/mklml
60
60
-DENABLE_OP_TIMER=${ANAKIN_ENABLE_OP_TIMER}
61
+ -DBUILD_FAT_BIN=${ANAKIN_BUILD_FAT_BIN}
62
+ -DBUILD_CROSS_PLANTFORM=${ANAKIN_BUILD_CROSS_PLANTFORM}
61
63
${EXTERNAL_OPTIONAL_ARGS}
62
64
CMAKE_CACHE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=${ANAKIN_INSTALL_DIR}
63
65
)
64
66
65
67
message (STATUS "Anakin for inference is enabled" )
66
68
message (STATUS "Anakin is set INCLUDE:${ANAKIN_INCLUDE} LIBRARY:${ANAKIN_LIBRARY} " )
67
-
69
+ add_dependencies ( extern_anakin protobuf mklml )
68
70
add_library (anakin_shared SHARED IMPORTED GLOBAL )
69
71
set_property (TARGET anakin_shared PROPERTY IMPORTED_LOCATION ${ANAKIN_SHARED_LIB} )
70
- add_dependencies (anakin_shared extern_anakin protobuf mklml )
72
+ add_dependencies (anakin_shared extern_anakin )
71
73
72
74
add_library (anakin_saber SHARED IMPORTED GLOBAL )
73
75
set_property (TARGET anakin_saber PROPERTY IMPORTED_LOCATION ${ANAKIN_SABER_LIB} )
74
- add_dependencies (anakin_saber extern_anakin protobuf mklml )
76
+ add_dependencies (anakin_saber extern_anakin )
75
77
76
78
list (APPEND external_project_dependencies anakin_shared anakin_saber )
Original file line number Diff line number Diff line change @@ -156,6 +156,8 @@ function cmake_gen() {
156
156
-DWITH_INFERENCE_API_TEST=${WITH_INFERENCE_API_TEST:- ON}
157
157
-DINFERENCE_DEMO_INSTALL_DIR=${INFERENCE_DEMO_INSTALL_DIR}
158
158
-DWITH_ANAKIN=${WITH_ANAKIN:- OFF}
159
+ -DANAKIN_BUILD_FAT_BIN=${ANAKIN_BUILD_FAT_BIN: OFF}
160
+ -DANAKIN_BUILD_CROSS_PLANTFORM=${ANAKIN_BUILD_CROSS_PLANTFORM: ON}
159
161
-DPY_VERSION=${PY_VERSION:- 2.7}
160
162
-DCMAKE_INSTALL_PREFIX=${INSTALL_PREFIX:-/ paddle/ build}
161
163
========================================
188
190
-DWITH_INFERENCE_API_TEST=${WITH_INFERENCE_API_TEST:- ON} \
189
191
-DINFERENCE_DEMO_INSTALL_DIR=${INFERENCE_DEMO_INSTALL_DIR} \
190
192
-DWITH_ANAKIN=${WITH_ANAKIN:- OFF} \
193
+ -DANAKIN_BUILD_FAT_BIN=${ANAKIN_BUILD_FAT_BIN: OFF} \
194
+ -DANAKIN_BUILD_CROSS_PLANTFORM=${ANAKIN_BUILD_CROSS_PLANTFORM: ON} \
191
195
-DPY_VERSION=${PY_VERSION:- 2.7} \
192
196
-DCMAKE_INSTALL_PREFIX=${INSTALL_PREFIX:-/ paddle/ build}
193
197
@@ -777,6 +781,17 @@ function main() {
777
781
test_fluid_lib
778
782
assert_api_spec_approvals
779
783
;;
784
+ assert_api)
785
+ assert_api_not_changed ${PYTHON_ABI:- " " }
786
+ ;;
787
+ test_inference)
788
+ gen_capi_package
789
+ gen_fluid_lib
790
+ test_fluid_lib
791
+ ;;
792
+ assert_api_approvals)
793
+ assert_api_spec_approvals
794
+ ;;
780
795
maccheck)
781
796
cmake_gen ${PYTHON_ABI:- " " }
782
797
build_mac
You can’t perform that action at this time.
0 commit comments