update

PaddlePaddle · Nov 27, 2023 · f4b2481 · f4b2481
1 parent ae1e15e
commit f4b2481
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 38 deletions.
diff --git a/paddle/fluid/framework/ir/conv2d_fusion_layout_transfer_pass.cc b/paddle/fluid/framework/ir/conv2d_fusion_layout_transfer_pass.cc
@@ -195,7 +195,7 @@ void Conv2dFusionLayoutTransferPass::ApplyImpl(ir::Graph *graph) const {
       auto *op_desc = op_node->Op();
 
       if (CutlassIsValid(op_node)) {
-        // conv2d_fusion_cutlass must have this attribute because of signature.
+        // conv2d_fusion must have this attribute because of signature.
         if (!op_desc->HasAttr("fuse_alpha")) {
           op_desc->SetAttr("fuse_alpha", 0.f);
         }

diff --git a/paddle/fluid/framework/ir/conv_elementwise_add_act_fuse_pass.cc b/paddle/fluid/framework/ir/conv_elementwise_add_act_fuse_pass.cc
@@ -208,7 +208,8 @@ void ConvElementwiseAddActFusePass::ApplyImpl(ir::Graph* graph) const {
         PrepareOpDesc(base_op_desc, bias_name, act_op_type, act_op_out, alpha);
     framework::OpDesc new_op_desc(new_op_proto, nullptr);
     if (cutlass_can_fuse && cutlass_enable && is_fp16_precision) {
-      new_op_desc.SetAttr("cudnn", false);
+      new_op_desc.SetAttr("use_cudnn", false);
+      new_op_desc.Flush();
     }
     // Create a new node for the fused op.
     auto* new_conv_op = graph->CreateOpNode(&new_op_desc);

diff --git a/paddle/phi/ops/compat/conv2d_sig.cc b/paddle/phi/ops/compat/conv2d_sig.cc