admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-02-02 01:06:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

withdraw pr2467 on branch 0.11.0 (#4002)

Browse Source 
Signed-off-by: cmli <chengming.li@zilliz.com>

Co-authored-by: cmli <chengming.li@zilliz.com>
This commit is contained in:
op-hunter 2020-10-15 19:39:49 +08:00 committed by GitHub
parent 4c23773e82
commit cdd40689f5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
core/src/index/thirdparty/faiss/gpu/utils/Select.cuh vendored
View File

@ -124,8 +124,16 @@ struct BlockSelect {
__device__ inline void addThreadQ(K k, V v) {
if (Dir ? Comp::gt(k, warpKTop) : Comp::lt(k, warpKTop)) {
threadK[numVals] = k;
threadV[numVals ++] = v;
// Rotate right
#pragma unroll
for (int i = NumThreadQ - 1; i > 0; --i) {
threadK[i] = threadK[i - 1];
threadV[i] = threadV[i - 1];
}
threadK[0] = k;
threadV[0] = v;
++numVals;
}
}