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
[CustomDevice]add custom NPU support for rnn #3521
[CustomDevice]add custom NPU support for rnn #3521
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
整体清理下PaddleNLP中npu的device选项,全部迁移到ascend,尽快完成标准化。
338aec8
to
bf8de72
Compare
目前Custom NPU还没有跑通bert和ernie,等跑通后再提PR清理这两个模型中的npu device选项。 |
@@ -266,6 +278,15 @@ python predict.py --vocab_path='./vocab.json' \ | |||
--params_path=checkpoints/final.pdparams | |||
``` | |||
|
|||
Custom NPU启动: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
建议改成Ascend NPU
因为NPU还是比较宽泛的,不一定特指昇腾
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done, THX!
@@ -181,10 +181,22 @@ python train.py --vocab_path='./vocab.json' \ | |||
--save_dir='./checkpoints' | |||
``` | |||
|
|||
Custom NPU 启动: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ascend NPU启动
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done, THX!
dc22015
to
d6410b8
Compare
PR types
New features
PR changes
Others
Description
add custom NPU support for rnn.