Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
Paddle-Lite
提交
b13ce36a
P
Paddle-Lite
项目概览
PaddlePaddle
/
Paddle-Lite
通知
331
Star
4
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
271
列表
看板
标记
里程碑
合并请求
78
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle-Lite
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
271
Issue
271
列表
看板
标记
里程碑
合并请求
78
合并请求
78
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
b13ce36a
编写于
9月 28, 2018
作者:
xiebaiyuan
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
suit ras berry compile cmake version
上级
b297bcbb
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
146 addition
and
77 deletion
+146
-77
test/CMakeLists.txt
test/CMakeLists.txt
+122
-67
tools/op.cmake
tools/op.cmake
+24
-10
未找到文件。
test/CMakeLists.txt
浏览文件 @
b13ce36a
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
()
tools/op.cmake
浏览文件 @
b13ce36a
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.
先完成此消息的编辑！
取消
想要评论请
注册
或
登录
