Skip to content
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

Open
wants to merge 2 commits into
base: branch-25.02
Choose a base branch
from

Conversation

wphicks
Copy link
Contributor

@wphicks wphicks commented Jan 23, 2025

Ensure that make_blobs respects globally-set device type.

@wphicks wphicks requested a review from a team as a code owner January 23, 2025 22:16
@wphicks wphicks requested review from teju85 and divyegala January 23, 2025 22:16
@github-actions github-actions bot added the Cython / Python Cython or Python issue label Jan 23, 2025
@wphicks wphicks added feature request New feature or request non-breaking Non-breaking change cuml-cpu labels Jan 23, 2025
Comment on lines 172 to 174
# 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
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

comment needs updating

Copy link
Member

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"
Copy link
Member

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cuml-cpu Cython / Python Cython or Python issue feature request New feature or request non-breaking Non-breaking change
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants