diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt index 1fb5fcf2df3bd26c094e6d79b37e7ba87d0e7475..5602d2d3ac38a7f9dfb175a0b03b1260960fc32e 100644 --- a/test/CMakeLists.txt +++ b/test/CMakeLists.txt @@ -129,6 +129,15 @@ if (CON GREATER -1) endif () +list(FIND NET "super" CON) +if (CON GREATER -1) + # gen test + ADD_EXECUTABLE(test-super net/test_super.cpp test_helper.h test_include.h executor_for_test.h) + target_link_libraries(test-super paddle-mobile) + set(FOUND_MATCH ON) + +endif () + if (NOT FOUND_MATCH) # gen test ADD_EXECUTABLE(test-resnet net/test_resnet.cpp test_helper.h test_include.h executor_for_test.h) diff --git a/tools/build.sh b/tools/build.sh index f83a7fe1876b014a12f0acfa8ddbef61d953ceda..5cb0887e3aa43038f1cc58c915f1e6809dd4ac89 100755 --- a/tools/build.sh +++ b/tools/build.sh @@ -1,6 +1,6 @@ #!/usr/bin/env bash NETS="" -declare -a supportedNets=("googlenet" "mobilenet" "yolo" "squeezenet" "resnet" "mobilenetssd" "nlp" "mobilenetfssd" "genet") +declare -a supportedNets=("googlenet" "mobilenet" "yolo" "squeezenet" "resnet" "mobilenetssd" "nlp" "mobilenetfssd" "genet" "super") build_for_mac() { if [ ! `which brew` ]; then diff --git a/tools/op.cmake b/tools/op.cmake index 4fe9988e55629feb65ecd49b5c3a72f6faf12997..005ca08968082c3b04a58ca4be5a603037acf676 100644 --- a/tools/op.cmake +++ b/tools/op.cmake @@ -197,6 +197,16 @@ if (CON GREATER -1) set(FOUND_MATCH ON) endif() +list(FIND NET "super" CON) +if (CON GREATER -1) + message("super enabled") + set(FUSION_CONVADD_OP ON) + set(FUSION_CONVADDRELU_OP ON) + set(ELEMENTWISEADD_OP ON) + + set(FOUND_MATCH ON) +endif() + if(NOT FOUND_MATCH) message("--default--")