fix pre-dispatch bug (#10869)

chen2016013 · phlrain · web-flow · commit e1efedd1783b · 2025-07-21T14:47:48.000+08:00
Co-authored-by: phlrain &lt;phliuhongyu@126.com&gt;
diff --git a/paddlenlp/transformers/token_dispatcher.py b/paddlenlp/transformers/token_dispatcher.py
@@ -258,6 +258,6 @@ def backward(self, token_probs_g):
             True,
             True,
         )
-        probs_reshape_g = paddle._C_ops.reshape_grad(probs_grad, self.probs)
+        probs_reshape_g = paddle._C_ops.reshape_grad(self.probs, probs_grad)
         self.reset_status()
         return probs_reshape_g

Original file line number	Diff line number	Diff line change
`@@ -258,6 +258,6 @@ def backward(self, token_probs_g):`
`258`	`258`	`True,`
`259`	`259`	`True,`
`260`	`260`	`)`
`261`		`- probs_reshape_g = paddle._C_ops.reshape_grad(probs_grad, self.probs)`
	`261`	`+ probs_reshape_g = paddle._C_ops.reshape_grad(self.probs, probs_grad)`
`262`	`262`	`self.reset_status()`
`263`	`263`	`return probs_reshape_g`