PaddlePaddle
diff --git a/‎paddle/fluid/eager/auto_code_generator/generator/eager_gen.py
Copy file name to clipboard
+24-20Lines changed: 24 additions & 20 deletions b/‎paddle/fluid/eager/auto_code_generator/generator/eager_gen.py
Copy file name to clipboard
+24-20Lines changed: 24 additions & 20 deletions
diff --git a/‎paddle/fluid/eager/auto_code_generator/generator/python_c_gen.py
Copy file name to clipboard
-1Lines changed: 0 additions & 1 deletion b/‎paddle/fluid/eager/auto_code_generator/generator/python_c_gen.py
Copy file name to clipboard
-1Lines changed: 0 additions & 1 deletion
diff --git a/‎python/paddle/distributed/auto_parallel/__init__.py
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions b/‎python/paddle/distributed/auto_parallel/__init__.py
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions
@@ -55,9 +55,6 @@
 
 # Black Ops list that's NO NEED to apply code generation
 black_ops_list = [
-    "conv2d",
-    "conv2d_grad",
-    "conv2d_grad_grad",
     "add_n",
     "add_n_grad",
     "sync_batch_norm_",
@@ -68,6 +65,8 @@
     "push_gpups_sparse",
 ]
 
+only_backward_ops_list = ["conv2d"]
+
 
 # white ops list whose kernel can be deleted after performance analysis
 # original kernel and its derivative kernel can be deleted when composite_grad
@@ -3248,10 +3247,14 @@ def GenerateCode(self, grad_flag=False):
         for forward_api_contents in true_forward_api_list:
             if forward_api_contents[op_string] in black_ops_list:
                 continue
-            if op_string == 'backward_op' and (
-                forward_api_contents[op_string].endswith(
-                    ('double_grad', 'triple_grad', 'grad_grad')
+            if (
+                op_string == 'backward_op'
+                and (
+                    forward_api_contents[op_string].endswith(
+                        ('double_grad', 'triple_grad', 'grad_grad')
+                    )
                 )
+                and "conv2d" not in forward_api_contents[op_string]
             ):
                 continue
 
@@ -3264,21 +3267,22 @@ def GenerateCode(self, grad_flag=False):
                     forward_api_contents
                 )
 
-            # Generate Dygraph Forward Function
-            function_generator = DygraphForwardFunctionGenerator(
-                forward_api_contents,
-                backward_api_contents,
-                forward_apis_dict,
-                namespace,
-            )
-            function_generator.run(grad_flag)
+            if forward_api_contents[op_string] not in only_backward_ops_list:
+                # Generate Dygraph Forward Function
+                function_generator = DygraphForwardFunctionGenerator(
+                    forward_api_contents,
+                    backward_api_contents,
+                    forward_apis_dict,
+                    namespace,
+                )
+                function_generator.run(grad_flag)
 
-            self.forward_definition_str += (
-                function_generator.forward_definition_str + "\n"
-            )
-            self.forward_declaration_str += (
-                function_generator.forward_declaration_str + "\n"
-            )
+                self.forward_definition_str += (
+                    function_generator.forward_definition_str + "\n"
+                )
+                self.forward_declaration_str += (
+                    function_generator.forward_declaration_str + "\n"
+                )
 
             if not grad_flag:
                 # Generate Dygraph GradNode Function
 
@@ -30,7 +30,6 @@
     "scale_grad",
     "push_gpups_sparse",
     "multiply_grad",
-    "conv2d_grad",
     "pull_sparse_v2_grad",
 }
 
 
@@ -24,6 +24,7 @@
 )
 from .process_mesh import ProcessMesh  # noqa: F401
 from .random import parallel_manual_seed  # noqa: F401
+from .ring_conv import RingConv2d  # noqa: F401
 from .static.engine import Engine  # noqa: F401
 from .strategy import Strategy  # noqa: F401
Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,6 @@`
`30`	`30`	`"scale_grad",`
`31`	`31`	`"push_gpups_sparse",`
`32`	`32`	`"multiply_grad",`
`33`		`- "conv2d_grad",`
`34`	`33`	`"pull_sparse_v2_grad",`
`35`	`34`	`}`
`36`	`35`
Original file line number	Diff line number	Diff line change
`@@ -24,6 +24,7 @@`
`24`	`24`	`)`
`25`	`25`	`from .process_mesh import ProcessMesh # noqa: F401`
`26`	`26`	`from .random import parallel_manual_seed # noqa: F401`
	`27`	`+from .ring_conv import RingConv2d # noqa: F401`
`27`	`28`	`from .static.engine import Engine # noqa: F401`
`28`	`29`	`from .strategy import Strategy # noqa: F401`
`29`	`30`