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

[npu] update npu api #3148

Merged
merged 3 commits into from
Apr 11, 2023
Merged

Conversation

YanhuiDua
Copy link
Contributor

PR types

Others

PR changes

APIs

Description

update is_compiled_with_npu to is_compiled_with_custim_device("npu")

Linked to PaddlePaddle/Paddle#52385

tools/predict.py Outdated
@@ -147,7 +147,7 @@ def main(args):
place = 'gpu'
elif args.device == 'xpu' and paddle.is_compiled_with_xpu():
place = 'xpu'
elif args.device == 'npu' and paddle.is_compiled_with_npu():
elif args.device == 'npu' and paddle.is_compilis_compiled_with_custom_device('npu')ed_with_npu():
Copy link
Collaborator

Choose a reason for hiding this comment

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

这里看下是否错误了

Copy link
Contributor Author

Choose a reason for hiding this comment

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

thx, done

@juncaipeng juncaipeng merged commit e36f77a into PaddlePaddle:release/2.7 Apr 11, 2023
@YanhuiDua YanhuiDua deleted the updpate_npu_api branch April 11, 2023 11:23
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