Skip to content

Compilation optimizations - batch 1 #5112

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

Merged
merged 1 commit into from
Jun 10, 2025

Conversation

ChuckHastings
Copy link
Collaborator

@ChuckHastings ChuckHastings commented Jun 9, 2025

Breaking previous PR into smaller chunks. Should be easier to diagnose issues causing failures of tests in CI.

Batch 1: add arithmetic variant types, first use in groupby and count functions.

2% reduction in size of libcugraph.so

@ChuckHastings ChuckHastings requested a review from a team as a code owner June 9, 2025 04:48
@ChuckHastings ChuckHastings self-assigned this Jun 9, 2025
@ChuckHastings ChuckHastings added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change and removed cuGraph labels Jun 9, 2025
{
return variant_type_dispatch(v, [](auto& v) {
using T = typename std::remove_reference<decltype(v)>::type::value_type;
return static_cast<arithmetic_device_span_t>(raft::device_span<T>(v.data(), v.size()));
Copy link
Contributor

Choose a reason for hiding this comment

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

Minor thing but isn't arithmetic_device_span_t(raft::device_span<T>(v.data(), v.size())) sufficient? And assigning a span to a variant of span might be more intuitive than statically casting a span to a variant of a span.

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

Yes, that would be better. I'll fix that in batch 2 of the changes.

@ChuckHastings
Copy link
Collaborator Author

/merge

@rapids-bot rapids-bot bot merged commit d3fa655 into rapidsai:branch-25.08 Jun 10, 2025
288 of 305 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
improvement Improvement / enhancement to an existing function non-breaking Non-breaking change
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants