-
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
[oneDNN] Cache oneDNN stream not to recreate in each oneDNN op #30358
Conversation
- compilation fix - fix - Fix - fix - added stream holder - compilation fix - Modified all remining files - fix - lint
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.
Please use auto&
to make sure that you're not doing copy while reaching for stream object.
@arogowie-intel I have added some additional change (last commit) to fix a problem exposed by test-analyzer-detect. If you want more details then ask me. |
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
@luotao1 Could you please start your review? CI will not pass without approval. This PR is changing way oneDNN stream was created , so due to API change I had do it all at once in all locations. So there many files modified, but most of them it is single line of change. |
PR types
Performance optimization
PR changes
OPs
Describe
created on dnnl::stream involves small memory allocation. stream can be created once and then reused so this PR makes stream to be created one per thread and store in per thread object.