test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-04 21:09:06 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

fix binary flat may retruns wrong distance when nq larger than 1 (#1760)

Browse Source 
Signed-off-by: shengjun.li <shengjun.li@zilliz.com>
This commit is contained in:
shengjun.li 2020-03-26 16:45:49 +08:00 committed by GitHub
parent c4b7be8ed6
commit e9f7422bc5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
core/src/index/thirdparty/faiss/utils/BinaryDistance.cpp vendored
View File

@ -209,7 +209,7 @@ void binary_distence_knn_mc(
}
}
}
for (size_t i = 0, ni = 0; i < n1; i++) {
for (size_t i = 0; i < n1; i++) {
size_t n_i = 0;
float *distances_i = distances + i * k;
int64_t *labels_i = labels + i * k;
@ -218,7 +218,7 @@ void binary_distence_knn_mc(
size_t match_index = t * n1 + i;
size_t copy_num = std::min(k - n_i, match_num[match_index]);
memcpy(labels_i + n_i, match_data + match_index * k, copy_num * sizeof(int64_t));
memset(distances + n_i, 0, copy_num * sizeof(int32_t));
memset(distances_i + n_i, 0, copy_num * sizeof(float));
n_i += copy_num;
}
for (; n_i < k; n_i++) {