Skip to content

[FEA] Add mdspan based raft::runtime::distance::pairwise_distance #1338

Open
@ahendriksen

Description

@ahendriksen

Is your feature request related to a problem? Please describe.
raft::runtime::distance::pairwise_distance currently takes pointers and integer dimensions instead of mdspan.

Describe the solution you'd like
I'd like raft::runtime::distance::pairwise_distance to also have an overload with mdspan arguments.

Describe alternatives you've considered
NA

Additional context
The tests in distance_base.cuh currently are based on mdspan. If they use the runtime api, the test and distance kernel code would be better separated. This would prevent a change deep in the distance kernels causing a rebuild of all the test objects. This would make iterative development faster.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions