|
|
@ -191,7 +191,9 @@ void profile(bool use_mkldnn = false) {
|
|
|
|
|
|
|
|
|
|
|
|
if (use_mkldnn) {
|
|
|
|
if (use_mkldnn) {
|
|
|
|
cfg.EnableMKLDNN();
|
|
|
|
cfg.EnableMKLDNN();
|
|
|
|
std::unordered_set<std::string> op_list = {"conv3d"};
|
|
|
|
// Enable all the mkldnn supported ops except conv3d in dam
|
|
|
|
|
|
|
|
std::unordered_set<std::string> op_list = {"softmax", "elementwise_add",
|
|
|
|
|
|
|
|
"relu"};
|
|
|
|
cfg.SetMKLDNNOp(op_list);
|
|
|
|
cfg.SetMKLDNNOp(op_list);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
@ -235,7 +237,9 @@ void compare(bool use_mkldnn = false) {
|
|
|
|
SetConfig(&cfg);
|
|
|
|
SetConfig(&cfg);
|
|
|
|
if (use_mkldnn) {
|
|
|
|
if (use_mkldnn) {
|
|
|
|
cfg.EnableMKLDNN();
|
|
|
|
cfg.EnableMKLDNN();
|
|
|
|
std::unordered_set<std::string> op_list = {"conv3d"};
|
|
|
|
// Enable all the mkldnn supported ops except conv3d in dam
|
|
|
|
|
|
|
|
std::unordered_set<std::string> op_list = {"softmax", "elementwise_add",
|
|
|
|
|
|
|
|
"relu"};
|
|
|
|
cfg.SetMKLDNNOp(op_list);
|
|
|
|
cfg.SetMKLDNNOp(op_list);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|