inplace addto pass

390f47c2 · sneaxiy · 60b86b2f · 390f47c2
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/framework/ir/memory_optimize_pass/inplace_addto_op_pass.cc ...ramework/ir/memory_optimize_pass/inplace_addto_op_pass.cc +2 -1

未找到文件。
--- a/paddle/fluid/framework/ir/memory_optimize_pass/inplace_addto_op_pass.cc
+++ b/paddle/fluid/framework/ir/memory_optimize_pass/inplace_addto_op_pass.cc
@@ -180,7 +180,8 @@ void InplaceAddToOpPass::Run(Graph *graph) const {

    // NOTE(zhiqiu): currently, only conv2d_grad supports addto strategy
    if (right_generated_op->Name() != "conv2d_grad" &&
-        right_generated_op->Name() != "resnet_unit_grad") {
+        right_generated_op->Name() != "resnet_unit_grad" && 
+	right_generated_op->Name() != "custom_fused_dense_grad") {
      continue;
    }