-
Notifications
You must be signed in to change notification settings - Fork 5.6k
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
[Dy2stat] Fix Memory Optimization in run_program_op and Add SimNet as Unit Test #25383
[Dy2stat] Fix Memory Optimization in run_program_op and Add SimNet as Unit Test #25383
Conversation
Thanks for your contribution! |
Thanks for your contribution! |
Thanks for your contribution! |
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
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
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 for the rest
for (const framework::OpDesc *op : all_ops) { | ||
if (!boost::algorithm::ends_with(op->Type(), "_grad")) { | ||
continue; | ||
} |
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 have a question: here we use _grad
to determine whether a op is grad_op
, but some op's grad_op is not with _grad
suffix, such arrary_write
's grad_op is array_read
.
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.
maybe also can use op_role
?
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.
Thank you for great suggestion. Done
4797135
PR types
Others
PR changes
Others
Describe
Add Similarity Net as unit test. During the unit test, we found three problems:
This PR fixes the 1. problem but modify codes for the 2. 3. problems to make PR smaller. I will fix those two problems in the next PR(s)