提交 de056fe0 编写于 作者: xiebaiyuan's avatar xiebaiyuan

add single build for genet

上级 43309a91
......@@ -55,6 +55,11 @@ elseif("mobilenetfssd" IN_LIST NET)
# gen test
ADD_EXECUTABLE(test-fssd net/test_mobilenet_025_fssd.cpp test_helper.h test_include.h)
target_link_libraries(test-fssd paddle-mobile)
elseif("genet" IN_LIST NET)
# gen test
ADD_EXECUTABLE(test-genet net/test_genet_combine.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test-genet paddle-mobile)
else ()
# gen test
......
#!/usr/bin/env bash
NETS=""
declare -a supportedNets=("googlenet" "mobilenet" "yolo" "squeezenet" "resnet" "mobilenetssd" "nlp" "mobilenetfssd")
declare -a supportedNets=("googlenet" "mobilenet" "yolo" "squeezenet" "resnet" "mobilenetssd" "nlp" "mobilenetfssd" "genet")
build_for_mac() {
if [ ! `which brew` ]; then
......
......@@ -148,6 +148,21 @@ if ("mobilenetfssd" IN_LIST NET)
set(FOUND_MATCH ON)
endif()
if ("genet" IN_LIST NET)
message("genet enabled")
set(FUSION_CONVADDPRELU_OP ON)
set(FUSION_CONVADDADDPRELU_OP ON)
set(FUSION_CONVADD_OP ON)
set(CONV_TRANSPOSE_OP ON)
set(FUSION_CONVADDRELU_OP ON)
set(ELEMENTWISEADD_OP ON)
set(PRELU_OP ON)
set(POOL_OP ON)
set(CONCAT_OP ON)
set(FOUND_MATCH ON)
endif()
if(NOT FOUND_MATCH)
message("--default--")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册