-
Notifications
You must be signed in to change notification settings - Fork 553
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
Support CPU execution for make_blobs #6253
base: branch-25.02
Are you sure you want to change the base?
Conversation
# Set the default output type to "cupy". This will be ignored if the user | ||
# has set `cuml.global_settings.output_type`. Only necessary for array | ||
# generation methods that do not take an array as input |
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.
comment needs updating
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.
What does the output type "array" mean? Does it mean "leave it as the type that it is" or will it convert to a cuml array?
order=order, | ||
) | ||
|
||
assert out.shape == (n_samples, n_features), "out shape mismatch" |
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.
Should we add an assert
to check that out
is a cupy/numpy array when the device type is gpu/cpu? At least that would be my expectation on how make_blobs
should work
Ensure that
make_blobs
respects globally-set device type.