From 52af0a603e4761af0113fd4f28eb5b3bbdf2e4e4 Mon Sep 17 00:00:00 2001
From: hlygit66666 <32728786+hlygit66666@users.noreply.github.com>
Date: Mon, 14 Feb 2022 20:06:20 +0800
Subject: [PATCH] Add Inplace addto pass and unittest. (#39433)

* add fuse_relu_depthwise_conv_pass unittest

* fix atol and rtol

* fix according to review

* Update test_dist_fuse_relu_depthwise_conv_pass.py

* add inplace_addto pass and unittest
---
 python/paddle/distributed/passes/cpp_pass.py  | 13 +++
 .../test_dist_inplace_addto_pass.py           | 96 +++++++++++++++++++
 2 files changed, 109 insertions(+)
 create mode 100644 python/paddle/fluid/tests/unittests/distributed_passes/test_dist_inplace_addto_pass.py

diff --git a/python/paddle/distributed/passes/cpp_pass.py b/python/paddle/distributed/passes/cpp_pass.py
index 481fdd50a4..4a4e5ecbbb 100644
--- a/python/paddle/distributed/passes/cpp_pass.py
+++ b/python/paddle/distributed/passes/cpp_pass.py
@@ -80,3 +80,16 @@ class FuseOptimizerPass(CPPPassWrapper):
 
     def _type(self):
         return PassType.FUSION_OPT
+
+
+@register_pass("inplace_addto_op")
+class InplaceAddtoOpPass(CPPPassWrapper):
+    def __init__(self):
+        super(InplaceAddtoOpPass, self).__init__()
+
+    @property
+    def cpp_name(self):
+        return "inplace_addto_op_pass"
+
+    def _type(self):
+        return PassType.CALC_OPT
diff --git a/python/paddle/fluid/tests/unittests/distributed_passes/test_dist_inplace_addto_pass.py b/python/paddle/fluid/tests/unittests/distributed_passes/test_dist_inplace_addto_pass.py
new file mode 100644
index 0000000000..32a6257a5f
--- /dev/null
+++ b/python/paddle/fluid/tests/unittests/distributed_passes/test_dist_inplace_addto_pass.py
@@ -0,0 +1,96 @@
+# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
+# 
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import os
+import paddle
+import paddle.distributed.fleet as fleet
+import numpy as np
+import paddle.nn as nn
+from paddle.distributed.passes import new_pass, PassManager
+import unittest
+from dist_pass_test_base import DistPassTestBase
+
+
+class DemoNet(nn.Layer):
+    def __init__(self):
+        super(DemoNet, self).__init__()
+
+        self.conv1 = nn.Conv2D(3, 3, (3, 3), padding=1, data_format="NHWC")
+        self.conv2 = nn.Conv2D(3, 3, (3, 3), padding=1, data_format="NHWC")
+        self.conv3 = nn.Conv2D(3, 3, (3, 3), padding=1, data_format="NHWC")
+        self.relu = nn.ReLU()
+
+    def forward(self, x):
+        out = self.conv1(x)
+        out = self.conv2(out) + self.conv3(out)
+        out = self.relu(out)
+        out = paddle.flatten(out, 1)
+        return out
+
+
+class TestInplaceAddtoPass(DistPassTestBase):
+    def init(self):
+        self.atol = 0.0
+        self.rtol = 0.0
+        paddle.fluid.set_flags({"FLAGS_max_inplace_grad_add": 8})
+
+    def get_model(self, place, batch_size=32, image_shape=[224, 224, 3]):
+        image = paddle.static.data(
+            shape=[batch_size] + image_shape, dtype='float32', name='image')
+
+        model = DemoNet()
+        pred_out = model(image)
+        loss = paddle.mean(pred_out)
+        optimizer = paddle.optimizer.Adam(learning_rate=1e-3)
+
+        dist_strategy = fleet.DistributedStrategy()
+        dist_strategy.fuse_all_reduce_ops = False
+        dist_strategy.without_graph_optimization = True
+
+        fleet.init(is_collective=True, strategy=dist_strategy)
+        optimizer = fleet.distributed_optimizer(optimizer)
+        optimizer.minimize(loss)
+
+        rank = paddle.distributed.get_rank()
+
+        def reader():
+            seed = int(os.environ.get("SEED", 0))
+            np.random.seed(seed + rank)
+            for _ in range(10):
+                image_np = np.random.random(size=image.shape).astype('float32')
+                yield image_np,
+
+        main_program = paddle.static.default_main_program()
+        startup_program = paddle.static.default_startup_program()
+        print(main_program)
+        return main_program, startup_program, [image], [loss], reader
+
+    def apply_passes(self, main_prog, startup_prog):
+        pass_manager = PassManager(
+            [new_pass("inplace_addto_op", {"use_cuda": True})])
+        pass_manager.apply([main_prog], [startup_prog])
+        print(pass_manager.names)
+
+        conv2d_grad_attr = []
+        for op in main_prog.global_block().ops:
+            if op.type == "conv2d_grad":
+                conv2d_grad_attr.append(op.desc.attr("use_addto"))
+        self.assertTrue(True in conv2d_grad_attr)
+
+    def test_inplace_addto(self):
+        self.check_main()
+
+
+if __name__ == "__main__":
+    unittest.main()
-- 
GitLab