add log_softmax_op_npu #35006
Merged
add log_softmax_op_npu #35006
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
|
|
Thanks for your contribution! |
juneweng
force-pushed
the
log_softmax_op
branch
from
cd38d83
to
f27a0dc
Compare
firestonelib
previously approved these changes
Aug 19, 2021
qili93
reviewed
Aug 31, 2021
|
|
||
| #include "paddle/fluid/operators/log_softmax_op.h" | ||
| #include "paddle/fluid/framework/tensor_util.h" | ||
| #include "paddle/fluid/operators/npu_op_runner.h" |
| ops::LogSoftmaxNPUKernel<paddle::platform::NPUDeviceContext, float>, | ||
| ops::LogSoftmaxNPUKernel<paddle::platform::NPUDeviceContext, double>, | ||
| // ops::LogSoftmaxNPUKernel<paddle::platform::NPUDeviceContext, int>, // | ||
| // used to debug |
| dx = dout - np.exp(out) * dout.copy().sum(axis=axis, keepdims=True).repeat( | ||
| axis_dim, axis=axis) | ||
| return dx | ||
|
|
There was a problem hiding this comment.
以上两个func与test_log_softmax相同，可以直接修改为以下代码，不需要重复实现
from test_log_softmax import ref_log_softmax, test_log_softmax
| globals()[cls_name] = TestLogSoftmaxAxis | ||
|
|
||
|
|
||
| for _typename in {'float32'}: |
There was a problem hiding this comment.
可以支持fp64和fp16的情况下，这里增加float64 / float16的数据类型的单测
There was a problem hiding this comment.
float64 / float16精度有问题，这里只注册了fp32算子
There was a problem hiding this comment.
已将注册部分的fp16/fp64删去。
qili93
approved these changes
Sep 3, 2021
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
New features
PR changes
OPs
Describe
晟腾log_softmax算子适配
算子调用成功截图：