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?

[aamijar]

To follow up here are the binary size reports. The libcuvs size increases by 6.06 MB

• (1209.31 MB) Nightly https://downloads.rapids.ai/nightly/cuvs/2025-07-08/23c6dc0/cuda12_x86_64.compile_lib.html
• (1215.37 MB) This PR https://downloads.rapids.ai/ci/cuvs/pull-request/971/d91ffdb/cuda12_x86_64.compile_lib.html

--- Added Files ---
[+] CMakeFiles/cuvs_objs.dir/src/distance/detail/pairwise_matrix/dispatch_dice_double_double_double_int.cu.o: 981.89 KB
[+] CMakeFiles/cuvs_objs.dir/src/distance/detail/pairwise_matrix/dispatch_dice_float_float_float_int.cu.o: 3.09 MB
[+] CMakeFiles/cuvs_objs.dir/src/distance/detail/pairwise_matrix/dispatch_dice_half_float_float_int.cu.o: 2.69 MB

[csadorf]

Is this still targeting 25.12?

[robertmaynard]

All new files added via this PR need to use the SPDX header format now

[aamijar]

This one is not targeting 25.12, and yep will update to add the new header format.

[aamijar]

I've removed the cutlass operator and the binary size looks like this now:

CUDA 12

• (1096.15 MB) Nightly https://downloads.rapids.ai/nightly/cuvs/2025-12-16/0df2781/cuda12_x86_64.compile_lib.html
• (1099.35 MB) This PR https://downloads.rapids.ai/ci/cuvs/pull-request/971/3e730b3/cuda12_x86_64.compile_lib.html

--- Added Files ---
[+] CMakeFiles/cuvs_objs.dir/src/distance/detail/pairwise_matrix/dispatch_dice_double_double_double_int.cu.o: 651.816 KB
[+] CMakeFiles/cuvs_objs.dir/src/distance/detail/pairwise_matrix/dispatch_dice_float_float_float_int.cu.o: 1.186 MB
[+] CMakeFiles/cuvs_objs.dir/src/distance/detail/pairwise_matrix/dispatch_dice_half_float_float_int.cu.o: 2.010 MB

CUDA 13

• (432.98 MB) Nightly https://downloads.rapids.ai/nightly/cuvs/2025-12-16/0df2781/cuda13_x86_64.compile_lib.html
• (434.78 MB) This PR https://downloads.rapids.ai/ci/cuvs/pull-request/971/3e730b3/cuda13_x86_64.compile_lib.html

--- Added Files ---
[+] CMakeFiles/cuvs_objs.dir/src/distance/detail/pairwise_matrix/dispatch_dice_double_double_double_int.cu.o: 454.032 KB
[+] CMakeFiles/cuvs_objs.dir/src/distance/detail/pairwise_matrix/dispatch_dice_float_float_float_int.cu.o: 770.896 KB
[+] CMakeFiles/cuvs_objs.dir/src/distance/detail/pairwise_matrix/dispatch_dice_half_float_float_int.cu.o: 1.143 MB