未验证 提交 636c6347 编写于 作者: X xiaoxiaohehe001 提交者: GitHub

fix_convfusion (#44226)

上级 b0c9f24a
......@@ -2243,8 +2243,12 @@ PDNode *patterns::PriorBox::operator()() {
return boxes_var;
}
#if CUDNN_VERSION >= 8000
std::unordered_set<std::string> conv_act_set(
{"identity", "relu", "sigmoid", "tanh"});
#else
std::unordered_set<std::string> conv_act_set({"identity", "relu"});
#endif
PDNode *patterns::ConvElementwiseaddAct::operator()(PDNode *conv_in) {
conv_in->AsInput();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册