diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt index f4dc1421a4cd0f1062e8ad1240caa237e58c9371..d68a8c1fb1a2cd0584d80d5afa8ed8f439d5d5d4 100644 --- a/test/CMakeLists.txt +++ b/test/CMakeLists.txt @@ -1,161 +1,215 @@ 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) + 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) + 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) # gen test - ADD_EXECUTABLE(test-mobilenet-combine net/test_mobilenet_combine.cpp test_helper.h test_include.h executor_for_test.h) + 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) + 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) + 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) + 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) + 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) - ADD_EXECUTABLE(test-resnet net/test_resnet.cpp test_helper.h test_include.h executor_for_test.h) + 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) - ADD_EXECUTABLE(test-resnet50 fpga/test_resnet50.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-resnet50 fpga/test_resnet50.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-resnet50 paddle-mobile) - ADD_EXECUTABLE(test-fpga-EW fpga/test_fpga_EW.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-fpga-EW fpga/test_fpga_EW.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-fpga-EW paddle-mobile) - ADD_EXECUTABLE(test-fpga-conv fpga/test_fpga_conv.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-fpga-conv fpga/test_fpga_conv.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-fpga-conv paddle-mobile) - ADD_EXECUTABLE(test-fpga-pooling fpga/test_fpga_pooling.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-fpga-pooling fpga/test_fpga_pooling.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-fpga-pooling paddle-mobile) - ADD_EXECUTABLE(test-fpga-bypass fpga/test_fpga_bypass.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-fpga-bypass fpga/test_fpga_bypass.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-fpga-bypass paddle-mobile) - ADD_EXECUTABLE(test-fpga-softmax fpga/test_fpga_softmax.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-fpga-softmax fpga/test_fpga_softmax.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-fpga-softmax paddle-mobile) - ADD_EXECUTABLE(test-fpga-concat fpga/test_fpga_concat.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-fpga-concat fpga/test_fpga_concat.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-fpga-concat paddle-mobile) - ADD_EXECUTABLE(test-tensor-quant fpga/test_tensor_quant.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-tensor-quant fpga/test_tensor_quant.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-tensor-quant paddle-mobile) - ADD_EXECUTABLE(test-fpga-concat-op fpga/test_concat_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-fpga-concat-op fpga/test_concat_op.cpp test_helper.h test_include.h) target_link_libraries(test-fpga-concat-op paddle-mobile) - ADD_EXECUTABLE(test-format-data fpga/test_format_data.cpp test_helper.h test_include.h) + 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) + 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) + 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) # gen test - ADD_EXECUTABLE(test-gru-op operators/test_gru_op.cpp test_helper.h test_include.h) + 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) + 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) + 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) + 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) # gen test - ADD_EXECUTABLE(test-squeezenet net/test_squeezenet.cpp test_helper.h test_include.h executor_for_test.h) + 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) # gen test - ADD_EXECUTABLE(test-yolo net/test_yolo.cpp test_helper.h test_include.h executor_for_test.h) + 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) + 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) # gen test - ADD_EXECUTABLE(test-googlenet net/test_googlenet.cpp test_helper.h test_include.h executor_for_test.h) + 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) # gen test - ADD_EXECUTABLE(test-conv-op operators/test_cov_op.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-conv-op operators/test_cov_op.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-conv-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-mul-op operators/test_mul_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-mul-op operators/test_mul_op.cpp test_helper.h test_include.h) target_link_libraries(test-mul-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-elementwiseadd-op operators/test_elementwise_add_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-elementwiseadd-op operators/test_elementwise_add_op.cpp test_helper.h test_include.h) target_link_libraries(test-elementwiseadd-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-concat-op operators/test_concat_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-concat-op operators/test_concat_op.cpp test_helper.h test_include.h) target_link_libraries(test-concat-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-lrn-op operators/test_lrn_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-lrn-op operators/test_lrn_op.cpp test_helper.h test_include.h) target_link_libraries(test-lrn-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-batchnorm-op operators/test_batchnorm_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-batchnorm-op operators/test_batchnorm_op.cpp test_helper.h test_include.h) target_link_libraries(test-batchnorm-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-priorbox-op operators/test_prior_box_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-priorbox-op operators/test_prior_box_op.cpp test_helper.h test_include.h) target_link_libraries(test-priorbox-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-boxcoder-op operators/test_box_coder_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-boxcoder-op operators/test_box_coder_op.cpp test_helper.h test_include.h) target_link_libraries(test-boxcoder-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-transpose-op operators/test_transpose_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-transpose-op operators/test_transpose_op.cpp test_helper.h test_include.h) target_link_libraries(test-transpose-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-multiclassnms-op operators/test_multiclass_nms_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-multiclassnms-op operators/test_multiclass_nms_op.cpp test_helper.h test_include.h) target_link_libraries(test-multiclassnms-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-reshape-op operators/test_reshape_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-reshape-op operators/test_reshape_op.cpp test_helper.h test_include.h) target_link_libraries(test-reshape-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-relu-op operators/test_relu_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-relu-op operators/test_relu_op.cpp test_helper.h test_include.h) target_link_libraries(test-relu-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-fc-op operators/test_fusion_fc_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-fc-op operators/test_fusion_fc_op.cpp test_helper.h test_include.h) target_link_libraries(test-fc-op paddle-mobile) # gen test log @@ -205,19 +259,19 @@ else () target_link_libraries(test-openmp paddle-mobile) # gen test - ADD_EXECUTABLE(test-mobilenetssd net/test_mobilenet+ssd.cpp test_helper.h test_include.h executor_for_test.h) + 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) - # gen test - ADD_EXECUTABLE(test-mobilenet-combine net/test_mobilenet_combine.cpp test_helper.h test_include.h executor_for_test.h) + # 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) # gen test - ADD_EXECUTABLE(test-genet net/test_genet_combine.cpp test_helper.h test_include.h executor_for_test.h) + 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) # gen test - ADD_EXECUTABLE(test-sigmoid operators/test_sigmoid_op.cpp test_include.h) + ADD_EXECUTABLE(test-sigmoid operators/test_sigmoid_op.cpp test_include.h) target_link_libraries(test-sigmoid paddle-mobile) # gen test @@ -225,42 +279,43 @@ else () target_link_libraries(test-depthwise-conv-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-mobilenet net/test_mobilenet.cpp test_helper.h test_include.h executor_for_test.h) + 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) # gen test - ADD_EXECUTABLE(test-conv-add-relu-op operators/test_conv_add_relu_op.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-conv-add-relu-op operators/test_conv_add_relu_op.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-conv-add-relu-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-conv-add-bn-relu-op operators/test_fusion_conv_add_bn_relu_op.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-conv-add-bn-relu-op operators/test_fusion_conv_add_bn_relu_op.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-conv-add-bn-relu-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-nlp net/test_nlp.cpp test_helper.h test_include.h executor_for_test.h) + 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) # gen test - ADD_EXECUTABLE(test-gru-op operators/test_gru_op.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-gru-op operators/test_gru_op.cpp test_helper.h test_include.h) target_link_libraries(test-gru-op paddle-mobile) # gen test - ADD_EXECUTABLE(test-inceptionv4 net/test_inceptionv4.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-inceptionv4 net/test_inceptionv4.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-inceptionv4 paddle-mobile) # gen test - ADD_EXECUTABLE(test-alexnet net/test_alexnet.cpp test_helper.h test_include.h executor_for_test.h) + ADD_EXECUTABLE(test-alexnet net/test_alexnet.cpp test_helper.h test_include.h executor_for_test.h) target_link_libraries(test-alexnet paddle-mobile) - ADD_EXECUTABLE(test-googlenetv1 net/test_googlenetv1_combine.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-googlenetv1 net/test_googlenetv1_combine.cpp test_helper.h test_include.h) target_link_libraries(test-googlenetv1 paddle-mobile) # gen test - ADD_EXECUTABLE(test-fssd net/test_mobilenet_025_fssd.cpp test_helper.h test_include.h) + ADD_EXECUTABLE(test-fssd net/test_mobilenet_025_fssd.cpp test_helper.h test_include.h) target_link_libraries(test-fssd paddle-mobile) #add_library(test-lib-size SHARED common/test_lib_size.h common/test_lib_size.cpp) -endif() + +endif () diff --git a/tools/op.cmake b/tools/op.cmake index 8488dd86bd497a1f8c7425b0525986750ff5ac86..6158a318140cd4befebb68434dc8ef53d1b7cd07 100644 --- a/tools/op.cmake +++ b/tools/op.cmake @@ -1,5 +1,10 @@ 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)