-
Notifications
You must be signed in to change notification settings - Fork 185
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
Add a stack to the statistics resource #1563
Add a stack to the statistics resource #1563
Conversation
cb5cd34
to
5544fd7
Compare
5544fd7
to
d6fd147
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.
This is awesome, great contribution. Can't wait to see how we can connect it to nvtx and nsys. I have a few comments / questions.
Co-authored-by: Mark Harris <783069+harrism@users.noreply.github.com>
…s_resource_counters_stack
…s_resource_counters_stack
Co-authored-by: Bradley Dice <bdice@bradleydice.com>
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.
Minor nits, a nice demonstration of the flexibility RMM offers.
current_mr = rmm.mr.get_current_device_resource() | ||
yield | ||
finally: | ||
if current_mr is not rmm.mr.get_current_device_resource(): |
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.
Yes, the definition of equality between memory resources is that they are interchangeable for the purposes of allocate/deallocate
pairs, which is not sufficient for the usage here.
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.
More doc improvements.
…s_resource_counters_stack
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.
Can't wait to see it working in nsys!
Co-authored-by: Mark Harris <783069+harrism@users.noreply.github.com>
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.
Thanks Mads!
/merge |
Description
In order to enable more fine-grained statistics, this PR adds a stack to
statistics_resource_adaptor
Closes #1332
Checklist