Brute force supports multiple metrics, but only a few are exposed through the all-neighbors api.
|
auto allowed_metrics = params.metric == cuvs::distance::DistanceType::L2Expanded || |
|
params.metric == cuvs::distance::DistanceType::L2SqrtExpanded || |
|
params.metric == cuvs::distance::DistanceType::CosineExpanded || |
|
params.metric == cuvs::distance::DistanceType::InnerProduct; |
|
RAFT_EXPECTS(allowed_metrics, |
|
"Distance metric for all-neighbors build with brute force should be L2Expanded, " |
|
"L2SqrtExpanded, CosineExpanded, or InnerProduct."); |
|
} |
