fix op_desc set attr bug (#50281)

2755507c · Yuanle Liu · GitHub · 9b3a41b1 · 2755507c
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/framework/ir/map_op_to_another_pass.cc paddle/fluid/framework/ir/map_op_to_another_pass.cc +1 -0

未找到文件。
--- a/paddle/fluid/framework/ir/map_op_to_another_pass.cc
+++ b/paddle/fluid/framework/ir/map_op_to_another_pass.cc
@@ -57,6 +57,7 @@ void MapOp2AnotherPass::ApplyImpl(ir::Graph* graph) const {
      if (groups > 1) {
 #if CUDNN_VERSION >= 8100
        op_desc->SetType(replaced_map[op_type]);
+        op_desc->RemoveAttr("use_cudnn");
        op_desc->SetAttr("use_cudnn", true);
 #endif
      }