未验证 提交 cdd40689 编写于 作者: O op-hunter 提交者: GitHub

withdraw pr2467 on branch 0.11.0 (#4002)

Signed-off-by: Ncmli <chengming.li@zilliz.com>
Co-authored-by: Ncmli <chengming.li@zilliz.com>
上级 4c23773e
......@@ -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;
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册