From e21d9dd48f490d6aba6690b6bceadcdb41452cdc Mon Sep 17 00:00:00 2001 From: Victor Lafargue Date: Wed, 24 Mar 2021 21:13:04 +0100 Subject: [PATCH] Modify default value for rowMajorIndex and rowMajorQuery in bf-knn (#173) This PR modifies the default value for `rowMajorIndex` and `rowMajorQuery` parameters in `brute_force_knn`. Authors: - Victor Lafargue (@viclafargue) - Corey J. Nolet (@cjnolet) Approvers: - Corey J. Nolet (@cjnolet) URL: https://github.com/rapidsai/raft/pull/173 --- cpp/include/raft/spatial/knn/knn.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cpp/include/raft/spatial/knn/knn.hpp b/cpp/include/raft/spatial/knn/knn.hpp index 066d1cff3a..eae5a3c8cc 100644 --- a/cpp/include/raft/spatial/knn/knn.hpp +++ b/cpp/include/raft/spatial/knn/knn.hpp @@ -53,7 +53,7 @@ using deviceAllocator = raft::mr::device::allocator; inline void brute_force_knn( raft::handle_t const &handle, std::vector &input, std::vector &sizes, int D, float *search_items, int n, int64_t *res_I, - float *res_D, int k, bool rowMajorIndex = false, bool rowMajorQuery = false, + float *res_D, int k, bool rowMajorIndex = true, bool rowMajorQuery = true, std::vector *translations = nullptr, distance::DistanceType metric = distance::DistanceType::L2Unexpanded, float metric_arg = 2.0f) {