Skip to content

Conversation

@aamijar
Copy link
Member

@aamijar aamijar commented Jun 3, 2025

Resolves #5129
Depends on rapidsai/cuvs#971
Adding new distance types in cuvs bloats the binary size.

@copy-pr-bot
Copy link

copy-pr-bot bot commented Jun 3, 2025

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@github-actions github-actions bot added the Cython / Python Cython or Python issue label Jun 3, 2025
@aamijar aamijar added non-breaking Non-breaking change improvement Improvement / enhancement to an existing function labels Jun 3, 2025
@aamijar aamijar changed the title Enable dice metric for UMAP Enable dice metric for UMAP and pairwise_distances Jun 4, 2025
@aamijar aamijar marked this pull request as ready for review June 4, 2025 01:04
@aamijar aamijar requested review from a team as code owners June 4, 2025 01:04
@aamijar aamijar requested review from divyegala and vyasr June 4, 2025 01:04
FORK rapidsai
PINNED_TAG branch-${CUML_BRANCH_VERSION_cuvs}
FORK aamijar
PINNED_TAG umap-dice-coefficient
Copy link
Member Author

Choose a reason for hiding this comment

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

Remove later

@aamijar aamijar requested a review from a team as a code owner June 4, 2025 22:22
@github-actions github-actions bot added the ci label Jun 4, 2025
@github-actions github-actions bot removed the ci label Jun 12, 2025
@aamijar
Copy link
Member Author

aamijar commented Sep 22, 2025

Blocked on constraints around cuvs binary size.

@csadorf
Copy link
Contributor

csadorf commented Nov 14, 2025

What's the plan for this PR?

@aamijar
Copy link
Member Author

aamijar commented Nov 14, 2025

Currently this is a P2+ priority but basically, we need to refactor the cuvs side PR to avoid using cutlass which bloats the binary size.

@aamijar aamijar changed the base branch from branch-25.08 to main November 14, 2025 23:50
@aamijar aamijar marked this pull request as draft November 14, 2025 23:51
@csadorf
Copy link
Contributor

csadorf commented Nov 14, 2025

Currently this is a P2+ priority but basically, we need to refactor the cuvs side PR to avoid using cutlass which bloats the binary size.

Is there already a cuvs issue/PR for this?

@aamijar
Copy link
Member Author

aamijar commented Nov 14, 2025

Yes rapidsai/cuvs#971, also in this PR description.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CMake CUDA/C++ Cython / Python Cython or Python issue 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.

[FEA] Support for Dice coefficient as a metric in UMAP

2 participants