【Auto Parallel】Add has_bias check in fused_linear_param_grad_add (#72483)

Function-Samuel · web-flow · commit a424eaeef0ce · 2025-04-27T11:37:22.000+08:00
diff --git a/paddle/phi/api/generator/dist_api_gen.py b/paddle/phi/api/generator/dist_api_gen.py
@@ -1912,6 +1912,11 @@ def generate_output_dist_attr_setting(self) -> str:
                         )
                     )
                 else:
+                    if (
+                        self.kernel['func'][0] == 'fused_linear_param_grad_add'
+                        and i == 1
+                    ):
+                        set_out_dist_attr_code += "\n    if (has_bias)"
                     set_out_dist_attr_code += (
                         SET_SINGLE_OUT_REPLICATED_DIST_ATTR_TEMPLATE.format(
                             out_name

Original file line number	Diff line number	Diff line change
`@@ -1912,6 +1912,11 @@ def generate_output_dist_attr_setting(self) -> str:`
`1912`	`1912`	`)`
`1913`	`1913`	`)`
`1914`	`1914`	`else:`
	`1915`	`+ if (`
	`1916`	`+ self.kernel['func'][0] == 'fused_linear_param_grad_add'`
	`1917`	`+ and i == 1`
	`1918`	`+ ):`
	`1919`	`+ set_out_dist_attr_code += "\n if (has_bias)"`
`1915`	`1920`	`set_out_dist_attr_code += (`
`1916`	`1921`	`SET_SINGLE_OUT_REPLICATED_DIST_ATTR_TEMPLATE.format(`
`1917`	`1922`	`out_name`