-
Notifications
You must be signed in to change notification settings - Fork 915
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
Remove cudf._lib.null_mask in favor of inlining pylibcudf #17440
Remove cudf._lib.null_mask in favor of inlining pylibcudf #17440
Conversation
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.
One suggestions to avoid a bitmask_or
.
with acquire_spill_lock(): | ||
_, has_null_group = plc.null_mask.bitmask_or( | ||
[col.to_pylibcudf(mode="read") for col in index._columns] | ||
) |
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.
FWIW, this can be computed without a kernel, because the columns know their null count, so this is equivalent to:
has_null_group = any(col.has_nulls() for col in index._columns)
/merge |
Description
Contributes to #17317
Checklist