Moving dice distance from raft to cuvs and support for half types

[aamijar]

Resolves #966

[copy-pr-bot]

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.

[aamijar]

Adding @rhdong as a reviewer since he worked on half types for the other distances #225

[cjnolet]

@aamijar I know this isn't your fault, but we're closely monitoring the cuVS binary size going forward and the pairwise distance APIs are by far the largest offender. Can you check the metrics logs that are produced in this PR to see how much this new distance affects the libcuvs binary size?

[aamijar]

Hi @cjnolet, sure where can I check the binary size/metrics for this pr?

[bdice]

@aamijar See the conda C++ build logs for the build metrics reports: https://github.com/rapidsai/cuvs/actions/runs/15453622193/job/43501422197#step:14:31

The HTML report here (https://downloads.rapids.ai/ci/cuvs/pull-request/971/59cb596/cuda12_x86_64.compile_lib.html) shows some data for the new files:

CMakeFiles/cuvs_objs.dir/src/distance/detail/pairwise_matrix/dispatch_dice_float_float_float_int.cu.o	8.331 s	3.280 MB
CMakeFiles/cuvs_objs.dir/src/distance/detail/pairwise_matrix/dispatch_dice_half_float_float_int.cu.o	8.622 s	2.857 MB
CMakeFiles/cuvs_objs.dir/src/distance/detail/pairwise_matrix/dispatch_dice_double_double_double_int.cu.o	8.796 s	1.106 MB

So at least 7 MB just from the new files. There may be other changes that affect the binary size of existing objects too. Maybe distance.cu?

CMakeFiles/cuvs_objs.dir/src/distance/distance.cu.o	13.587 s	6.872 MB

[aamijar]

Thank you @bdice! It would be nice if there was a way to highlight in the html report which files are edited from the PR and the increase or decrease in binary size.

[rhdong]

LGTM

[divyegala]

@aamijar I suspect the binary size is higher because we are using cutlass. @cjnolet given that this will be an infrequently used distance type, can we skip using cutlass and just use our normal pairwise kernel?