[MLU] Fix add grad bug (#1352)

wangna11BD · web-flow · commit 34c1d4059c7c · 2024-07-19T16:15:58.000+08:00
diff --git a/backends/mlu/kernels/elementwise_add_kernel.cc b/backends/mlu/kernels/elementwise_add_kernel.cc
@@ -73,7 +73,7 @@ void AddGradKernel(const Context& dev_ctx,
                       dx_desc.get(),
                       GetBasePtr(dx));
     } else {
-      *dx = dout;
+      TensorCopy(dev_ctx, dout, false, dx);
     }
   }
   if (dy) {
@@ -104,7 +104,7 @@ void AddGradKernel(const Context& dev_ctx,
                       dy_desc.get(),
                       GetBasePtr(dy));
     } else {
-      *dy = dout;
+      TensorCopy(dev_ctx, dout, false, dy);
     }
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@ void AddGradKernel(const Context& dev_ctx,`
`73`	`73`	`dx_desc.get(),`
`74`	`74`	`GetBasePtr(dx));`
`75`	`75`	`} else {`
`76`		`- *dx = dout;`
	`76`	`+ TensorCopy(dev_ctx, dout, false, dx);`
`77`	`77`	`}`
`78`	`78`	`}`
`79`	`79`	`if (dy) {`
`@@ -104,7 +104,7 @@ void AddGradKernel(const Context& dev_ctx,`
`104`	`104`	`dy_desc.get(),`
`105`	`105`	`GetBasePtr(dy));`
`106`	`106`	`} else {`
`107`		`- *dy = dout;`
	`107`	`+ TensorCopy(dev_ctx, dout, false, dy);`
`108`	`108`	`}`
`109`	`109`	`}`
`110`	`110`	`}`