fix grad clip in xpu

Waynezee · Waynezee · commit 697a980f8743 · 2025-03-05T14:35:36.000+08:00
diff --git a/python/paddle/nn/clip.py b/python/paddle/nn/clip.py
@@ -773,7 +773,7 @@ def async_add_n(var_list):
         global_norm_var = async_add_n(global_norm_var)
         global_norm_var = paddle.sqrt(global_norm_var)
         max_global_norm = paddle.full(
-            shape=[], dtype=sum_dtype, fill_value=self.clip_norm
+            shape=[1], dtype=sum_dtype, fill_value=self.clip_norm
         )
 
         need_clip = False

Original file line number	Diff line number	Diff line change
`@@ -773,7 +773,7 @@ def async_add_n(var_list):`
`773`	`773`	`global_norm_var = async_add_n(global_norm_var)`
`774`	`774`	`global_norm_var = paddle.sqrt(global_norm_var)`
`775`	`775`	`max_global_norm = paddle.full(`
`776`		`- shape=[], dtype=sum_dtype, fill_value=self.clip_norm`
	`776`	`+ shape=[1], dtype=sum_dtype, fill_value=self.clip_norm`
`777`	`777`	`)`
`778`	`778`
`779`	`779`	`need_clip = False`