Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add NPU support for zero_copy_tensor. #34629

Merged
merged 4 commits into from
Aug 5, 2021

Conversation

houj04
Copy link
Contributor

@houj04 houj04 commented Aug 5, 2021

PR types

Bug fixes

PR changes

Others

Describe

ZeroCopyTensor的基类Tensor,在copy_to_cpu和copy_from_cpu,在NPU环境下会有问题,因为原有的代码里面少写了NPU的分支。

所以简单修改了一把,单测的修改也只是简单的加了两行。

@paddle-bot-old
Copy link

paddle-bot-old bot commented Aug 5, 2021

Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

Copy link
Contributor

@jiweibo jiweibo left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@jiweibo jiweibo merged commit a68709d into PaddlePaddle:develop Aug 5, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants