From 390f47c2b590d4c77b5db22962f51f903da683e1 Mon Sep 17 00:00:00 2001
From: sneaxiy <sneaxiy@126.com>
Date: Fri, 11 Mar 2022 11:23:12 +0000
Subject: [PATCH] inplace addto pass

---
 .../framework/ir/memory_optimize_pass/inplace_addto_op_pass.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git 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
index 0ed2ec51b89..acbf951ef5d 100644
--- 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;
     }
 
-- 
GitLab