diff --git a/cpp/tests/neighbors/brute_force_prefiltered.cu b/cpp/tests/neighbors/brute_force_prefiltered.cu index aac00f39af..93c62ef5c9 100644 --- a/cpp/tests/neighbors/brute_force_prefiltered.cu +++ b/cpp/tests/neighbors/brute_force_prefiltered.cu @@ -1012,21 +1012,23 @@ const std::vector> selectk_inputs = { {1024, 8192, 3, 1, 0.1, cuvs::distance::DistanceType::InnerProduct}, {1024, 8192, 5, 1, 0.1, cuvs::distance::DistanceType::L2SqrtExpanded}, {1024, 8192, 8, 1, 0.1, cuvs::distance::DistanceType::CosineExpanded}, - {1024, 8192, 2050, 16, 0.4, cuvs::distance::DistanceType::L2Expanded}, - {1024, 8192, 2051, 16, 0.5, cuvs::distance::DistanceType::L2Expanded}, - {1024, 8192, 2052, 16, 0.2, cuvs::distance::DistanceType::L2Expanded}, - {1024, 8192, 2050, 16, 0.4, cuvs::distance::DistanceType::InnerProduct}, - {1024, 8192, 2051, 16, 0.5, cuvs::distance::DistanceType::InnerProduct}, - {1024, 8192, 2052, 16, 0.2, cuvs::distance::DistanceType::InnerProduct}, + {1024, 8192, 2050, 16, 0.04, cuvs::distance::DistanceType::L2Expanded}, + {1024, 8192, 2051, 16, 0.05, cuvs::distance::DistanceType::L2Expanded}, + {1024, 8192, 2052, 16, 0.035, cuvs::distance::DistanceType::L2Expanded}, - {1024, 8192, 2050, 16, 0.4, cuvs::distance::DistanceType::L2SqrtExpanded}, - {1024, 8192, 2051, 16, 0.5, cuvs::distance::DistanceType::L2SqrtExpanded}, - {1024, 8192, 2052, 16, 0.2, cuvs::distance::DistanceType::L2SqrtExpanded}, - {1024, 8192, 2050, 16, 0.4, cuvs::distance::DistanceType::CosineExpanded}, - {1024, 8192, 2051, 16, 0.5, cuvs::distance::DistanceType::CosineExpanded}, + {1024, 8192, 2050, 16, 0.04, cuvs::distance::DistanceType::InnerProduct}, + {1024, 8192, 2051, 16, 0.05, cuvs::distance::DistanceType::InnerProduct}, + {1024, 8192, 2052, 16, 0.035, cuvs::distance::DistanceType::InnerProduct}, + + {1024, 8192, 2050, 16, 0.04, cuvs::distance::DistanceType::L2SqrtExpanded}, + {1024, 8192, 2051, 16, 0.05, cuvs::distance::DistanceType::L2SqrtExpanded}, + {1024, 8192, 2052, 16, 0.035, cuvs::distance::DistanceType::L2SqrtExpanded}, + + {1024, 8192, 2050, 16, 0.04, cuvs::distance::DistanceType::CosineExpanded}, + {1024, 8192, 2051, 16, 0.05, cuvs::distance::DistanceType::CosineExpanded}, + {1024, 8192, 2052, 16, 0.035, cuvs::distance::DistanceType::CosineExpanded}, - {1024, 8192, 2052, 16, 0.2, cuvs::distance::DistanceType::CosineExpanded}, {1024, 8192, 1, 16, 0.5, cuvs::distance::DistanceType::L2Expanded}, {1024, 8192, 2, 16, 0.2, cuvs::distance::DistanceType::L2Expanded},