-
Notifications
You must be signed in to change notification settings - Fork 540
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
Execution device interoperability documentation #5130
Execution device interoperability documentation #5130
Conversation
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
Looks good! Nice and concise. Had a few suggestions for the text, but the code looks good. I'd also be interested in any feedback from @beckernick if you have a moment to glance at this? |
This looks great overall. Left two small comments on the notebook. I think the small README update is the right choice until we add support for a few more operators and graduate device selection out of experimental. At that point, we can do a broader update. I think we should update the documentation with a device selection support matrix and update operator docstrings to indicate whether the operator supports device selection. I will separate issues for those and can take them on. |
The implication of this notebook is that it's possible (for all models) to:
Does this work for all models or only GLMs at the moment? I had thought the latter, and if so we may want to tweak the framing here to focus on the non-deployment scenario. |
You are right. Some of the models implement pickling and CPU/GPU interoperability but lack the CPU-to-GPU feature at the moment (i.e.: UMAP and HDBSCAN). Most models with serialization to disk + CPU/GPU interop should allow the deployment scenario though. Sure, I will modify the notebook so that it focuses a bit more on the non-deployment scenario and will also specify in the markdown that some of the models do not implement the CPU-to-GPU feature. |
Codecov ReportBase: 67.11% // Head: 67.20% // Increases project coverage by
Additional details and impacted files@@ Coverage Diff @@
## branch-23.02 #5130 +/- ##
================================================
+ Coverage 67.11% 67.20% +0.08%
================================================
Files 192 192
Lines 12396 12372 -24
================================================
- Hits 8320 8315 -5
+ Misses 4076 4057 -19
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. ☔ View full report at Codecov. |
def get_current_device_type(): | ||
return GlobalSettings().device_type |
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.
Is the reason for naming this function get_current_device_type()
as opposed to get_global_device_type()
, because the device type could be temporarily changed? If not, I would probably recommend to use a symmetric naming scheme here. Tagging @wphicks since I think you proposed the name in the first place.
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 for noticing this. Just corrected it.
a8d058c
to
a40befe
Compare
/merge |
Authors: - Victor Lafargue (https://github.com/viclafargue) - Dante Gama Dessavre (https://github.com/dantegd) Approvers: - Carl Simon Adorf (https://github.com/csadorf) - Dante Gama Dessavre (https://github.com/dantegd) URL: rapidsai#5130
No description provided.