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

suit ras berry compile cmake version

上级 b297bcbb
set(dir ${CMAKE_CURRENT_SOURCE_DIR})
set(CMAKE_RUNTIME_OUTPUT_DIRECTORY "${dir}/build")
set(FOUND_MATCH OFF)
if ("googlenet" IN_LIST NET)
set(CON -1)
message(STATUS "nets :${NET}")
list(FIND NET "googlenet" CON)
if (CON GREATER -1)
# gen test
ADD_EXECUTABLE(test-googlenet net/test_googlenet.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test-googlenet paddle-mobile)
elseif ("mobilenet" IN_LIST NET)
set(FOUND_MATCH ON)
endif ()
list(FIND NET "mobilenet" CON)
if (CON GREATER -1)
# gen test
ADD_EXECUTABLE(test-mobilenet net/test_mobilenet.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test-mobilenet paddle-mobile)
......@@ -13,23 +25,42 @@ elseif ("mobilenet" IN_LIST NET)
# gen test
ADD_EXECUTABLE(test-mobilenet-combine net/test_mobilenet_combine.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test-mobilenet-combine paddle-mobile)
set(FOUND_MATCH ON)
endif ()
elseif ("yolo" IN_LIST NET)
list(FIND NET "yolo" CON)
if (CON GREATER -1)
# gen test
ADD_EXECUTABLE(test-yolo net/test_yolo.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test-yolo paddle-mobile)
# gen test
ADD_EXECUTABLE(test_yolo_combined net/test_yolo_combined.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test_yolo_combined paddle-mobile)
elseif ("squeezenet" IN_LIST NET)
set(FOUND_MATCH ON)
endif ()
list(FIND NET "squeezenet" CON)
if (CON GREATER -1)
# gen test
ADD_EXECUTABLE(test-squeezenet net/test_squeezenet.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test-squeezenet paddle-mobile)
elseif("resnet" IN_LIST NET)
set(FOUND_MATCH ON)
endif ()
list(FIND NET "resnet" CON)
if (CON GREATER -1)
# gen test
ADD_EXECUTABLE(test-resnet net/test_resnet.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test-resnet paddle-mobile)
elseif("FPGAnets" IN_LIST NET)
set(FOUND_MATCH ON)
endif ()
list(FIND NET "FPGAnets" CON)
if (CON GREATER -1)
ADD_EXECUTABLE(test-resnet net/test_resnet.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test-resnet paddle-mobile)
......@@ -62,12 +93,22 @@ elseif("FPGAnets" IN_LIST NET)
ADD_EXECUTABLE(test-format-data fpga/test_format_data.cpp test_helper.h test_include.h)
target_link_libraries(test-format-data paddle-mobile)
elseif("mobilenetssd" IN_LIST NET)
set(FOUND_MATCH ON)
endif ()
list(FIND NET "mobilenetssd" CON)
if (CON GREATER -1)
# gen test
ADD_EXECUTABLE(test-mobilenetssd net/test_mobilenet+ssd.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test-mobilenetssd paddle-mobile)
elseif("nlp" IN_LIST NET)
set(FOUND_MATCH ON)
endif ()
list(FIND NET "nlp" CON)
if (CON GREATER -1)
# gen test
ADD_EXECUTABLE(test-nlp net/test_nlp.cpp test_helper.h test_include.h executor_for_test.h)
target_link_libraries(test-nlp paddle-mobile)
......@@ -75,17 +116,30 @@ elseif("nlp" IN_LIST NET)
# gen test
ADD_EXECUTABLE(test-gru-op operators/test_gru_op.cpp test_helper.h test_include.h)
target_link_libraries(test-gru-op paddle-mobile)
elseif("mobilenetfssd" IN_LIST NET)
set(FOUND_MATCH ON)
endif ()
list(FIND NET "mobilenetfssd" CON)
if (CON GREATER -1)
# 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)
set(FOUND_MATCH ON)
endif ()
list(FIND NET "genet" CON)
if (CON GREATER -1)
# 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 ()
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)
target_link_libraries(test-resnet paddle-mobile)
......@@ -263,4 +317,5 @@ else ()
#add_library(test-lib-size SHARED common/test_lib_size.h common/test_lib_size.cpp)
endif()
endif ()
set(FOUND_MATCH OFF)
if ("googlenet" IN_LIST NET)
set(CON -1)
message(STATUS "nets :${NET}")
list(FIND NET "googlenet" CON)
if (CON GREATER -1)
message("googlenet enabled")
set(CONCAT_OP ON)
set(CONV_OP ON)
......@@ -15,7 +20,8 @@ if ("googlenet" IN_LIST NET)
set(FOUND_MATCH ON)
endif()
if ("mobilenet" IN_LIST NET)
list(FIND NET "mobilenet" CON)
if (CON GREATER -1)
message("mobilenet enabled")
set(CONV_OP ON)
set(ELEMENTWISEADD_OP ON)
......@@ -33,7 +39,8 @@ if ("mobilenet" IN_LIST NET)
endif()
if ("mobilenetssd" IN_LIST NET)
list(FIND NET "mobilenetssd" CON)
if (CON GREATER -1)
message("mobilenetssd enabled")
set(FUSION_CONVBNRELU_OP ON)
set(FUSION_CONVBNRELU_OP ON)
......@@ -55,7 +62,8 @@ if ("mobilenetssd" IN_LIST NET)
endif()
if ("yolo" IN_LIST NET)
list(FIND NET "yolo" CON)
if (CON GREATER -1)
message("yolo enabled")
set(BATCHNORM_OP ON)
set(CONV_OP ON)
......@@ -65,7 +73,8 @@ if ("yolo" IN_LIST NET)
set(FOUND_MATCH ON)
endif()
if ("squeezenet" IN_LIST NET)
list(FIND NET "squeezenet" CON)
if (CON GREATER -1)
message("squeezenet enabled")
set(CONCAT_OP ON)
set(CONV_OP ON)
......@@ -79,7 +88,8 @@ if ("squeezenet" IN_LIST NET)
endif()
if ("resnet" IN_LIST NET)
list(FIND NET "resnet" CON)
if (CON GREATER -1)
message("resnet enabled")
set(CONCAT_OP ON)
set(CONV_OP ON)
......@@ -95,7 +105,8 @@ if ("resnet" IN_LIST NET)
set(FOUND_MATCH ON)
endif()
if ("FPGAnets" IN_LIST NET)
list(FIND NET "FPGAnets" CON)
if (CON GREATER -1)
message("FPGAnets enabled")
set(FUSION_CONVADDRELU_OP ON)
set(FUSION_CONVADDBNRELU_OP ON)
......@@ -114,7 +125,8 @@ if ("FPGAnets" IN_LIST NET)
set(FOUND_MATCH ON)
endif()
if ("nlp" IN_LIST NET)
list(FIND NET "nlp" CON)
if (CON GREATER -1)
message("nlp enabled")
set(FUSION_FC_OP ON)
set(LOOKUP_OP ON)
......@@ -127,7 +139,8 @@ if ("nlp" IN_LIST NET)
set(FOUND_MATCH ON)
endif()
if ("mobilenetfssd" IN_LIST NET)
list(FIND NET "mobilenetfssd" CON)
if (CON GREATER -1)
message("mobilenetfssd enabled")
set(FUSION_CONVADDRELU_OP ON)
set(FUSION_CONVADDBNRELU_OP ON)
......@@ -148,7 +161,8 @@ if ("mobilenetfssd" IN_LIST NET)
set(FOUND_MATCH ON)
endif()
if ("genet" IN_LIST NET)
list(FIND NET "genet" CON)
if (CON GREATER -1)
message("genet enabled")
set(FUSION_CONVADDPRELU_OP ON)
set(FUSION_CONVADDADDPRELU_OP ON)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册