fix_convfusion (#44226)

636c6347 · xiaoxiaohehe001 · GitHub · b0c9f24a · 636c6347
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

paddle/fluid/framework/ir/graph_pattern_detector.cc paddle/fluid/framework/ir/graph_pattern_detector.cc +4 -0

未找到文件。
--- a/paddle/fluid/framework/ir/graph_pattern_detector.cc
+++ b/paddle/fluid/framework/ir/graph_pattern_detector.cc
@@ -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();