-
Notifications
You must be signed in to change notification settings - Fork 5.7k
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
[NewIR] No.7 Migrate paddle.softmaxinto pir #57415
Conversation
看ci的报错,应该是新ir模式的切换导致了动静态图模式的混乱导致了单测的报错,需要在test_softmax_mkldnn_op的TestSoftmaxMKLDNNPrimitivesAlreadyExist.setUp中调用一下paddle.enable_static() |
@@ -1200,7 +1200,7 @@ def softmax(x, axis=-1, dtype=None, name=None): | |||
|
|||
if (dtype is not None) and (not isinstance(dtype, core.VarDesc.VarType)): |
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.
这里需要加上对于新IR datatype的检查
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~辛苦有空再review下
f086ac4
to
49cda24
Compare
49cda24
to
577ac42
Compare
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.
LGTM
if ( | ||
(dtype is not None) | ||
and (not isinstance(dtype, core.VarDesc.VarType)) | ||
and (not isinstance(dtype, core.DataType)) |
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.
and (not isinstance(dtype, core.DataType)) | |
and (not isinstance(dtype, (core.DataType, core.VarDesc.VarType))) |
PR types
Others
PR changes
Others
Description
No.7 Migrate paddle.softmax into pir