-
Notifications
You must be signed in to change notification settings - Fork 539
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
[REVIEW] Improved Array Conversion with CumlArrayDescriptor and Decorators [skip-ci] #3040
[REVIEW] Improved Array Conversion with CumlArrayDescriptor and Decorators [skip-ci] #3040
Conversation
…os where a function returns CumlArray
…mlarray-use-improved
…allow setting Base values in constructor
…mlarray-use-improved
…mlarray-use-improved
…mlarray-use-improved
…mlarray-use-improved
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.
Some minor comments, I have pretty much finished a first pass review
Co-authored-by: Dante Gama Dessavre <[email protected]>
…t-nv/cuml into bug-audit-cumlarray-use-improved
Co-authored-by: Dante Gama Dessavre <[email protected]>
…mlarray-use-improved
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.
Looking good! Let's discuss merge timing tomorrow. Just small comments 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.
Super small comments and questions, overall it looks great!
rerun tests |
Due to the large number of input and output array types that cuml supports, the library needs to be flexible in how array data is handed when entering and exiting the API. This has led to some inconsistent usage of
CumlArray
, unnecessary conversions and memcpys, and adds repeated work for the developers.This PR helps with these issues by adding a new object
CumlArrayDescriptor
and a set of decorators incuml.internals
that make working with array data easier for developers and more consistent for users, while still being flexible enough to handle any scenario. Finally, this PR updates the majority of the code base to be consistent and use the new features.A detailed description of the new features, how to use them, and whats going on internally, can be found in the new Estimator Guide.
Note: While the number of changed files is very large, the meat of the new features is limited to 2 sections. The rest of the changes are updates to use the new features and are very repetitive and formulaic.