未验证 提交 c9e7c407 编写于 作者: C Chen Weihang 提交者: GitHub

[Phi] Polish assign kernel copy impl (#43061)

* fix assign kernel copy impl

* fix test failed
上级 172739d4
...@@ -26,7 +26,7 @@ template <typename Context> ...@@ -26,7 +26,7 @@ template <typename Context>
void AssignKernel(const Context& dev_ctx, void AssignKernel(const Context& dev_ctx,
const DenseTensor& x, const DenseTensor& x,
DenseTensor* out) { DenseTensor* out) {
Copy<Context>(dev_ctx, x, x.place(), false, out); paddle::framework::TensorCopy(x, x.place(), out);
} }
template <typename Context> template <typename Context>
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册