-
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
Unify REGISTER_OP and REGISTER_OPERATOR #9911
Unify REGISTER_OP and REGISTER_OPERATOR #9911
Conversation
264652a
to
d9a60b4
Compare
d9a60b4
to
411e888
Compare
@@ -558,95 +558,126 @@ Swish Activation Operator. | |||
|
|||
namespace ops = paddle::operators; | |||
|
|||
REGISTER_OP(sigmoid, ops::ActivationOp, ops::SigmoidOpMaker, sigmoid_grad, | |||
ops::ActivationOpGrad); | |||
REGISTER_OPERATOR(sigmoid, ops::ActivationOp, ops::SigmoidOpMaker, |
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.
one suggestion, add ;
at the end of the register macro.
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.
I will do it in a separate PR, where I will grep all the op files.
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! Please add ;
in another PR
Awsome! It is helpful to optimize memory usage and prevent hanging when running |
fix #7158