update

Eddie-Wang1120 · Eddie-Wang1120 · commit 510f7757c075 · 2024-06-19T23:49:25.000+08:00
diff --git a/paddle/fluid/primitive/rule/vjp/details.h b/paddle/fluid/primitive/rule/vjp/details.h
@@ -798,7 +798,7 @@ void log_grad(const Tensor& x, const Tensor& out_grad, Tensor* x_grad) {
 template <typename T>
 void square_grad(const Tensor& x, const Tensor& out_grad, Tensor* x_grad) {
   if (x_grad) {
-    auto x_grad_tmp = 2 * x * out_grad;
+    Tensor x_grad_tmp = 2 * x * out_grad;
     set_output<T>(x_grad_tmp, x_grad);
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -798,7 +798,7 @@ void log_grad(const Tensor& x, const Tensor& out_grad, Tensor* x_grad) {`
`798`	`798`	`template <typename T>`
`799`	`799`	`void square_grad(const Tensor& x, const Tensor& out_grad, Tensor* x_grad) {`
`800`	`800`	`if (x_grad) {`
`801`		`- auto x_grad_tmp = 2 * x * out_grad;`
	`801`	`+ Tensor x_grad_tmp = 2 * x * out_grad;`
`802`	`802`	`set_output<T>(x_grad_tmp, x_grad);`
`803`	`803`	`}`
`804`	`804`	`}`