提交 98c9ead1 authored 作者: Adam Becker's avatar Adam Becker

fix bit shift bug

上级 bdc1bde9
...@@ -99,8 +99,8 @@ KERNEL void k_topk_dense_large( ...@@ -99,8 +99,8 @@ KERNEL void k_topk_dense_large(
#pragma unroll #pragma unroll
for (int bin=RADIX_SIZE-1; bin>=0; --bin) { for (int bin=RADIX_SIZE-1; bin>=0; --bin) {
if (smem[bin] >= k2) { if (smem[bin] >= k2) {
known_bits |= (bin << i); known_bits |= (((radix_t)bin) << i);
known_bits_mask |= ((RADIX_SIZE-1) << i); known_bits_mask |= (((radix_t)(RADIX_SIZE-1)) << i);
break; break;
} else } else
k2 -= smem[bin]; k2 -= smem[bin];
...@@ -111,8 +111,8 @@ KERNEL void k_topk_dense_large( ...@@ -111,8 +111,8 @@ KERNEL void k_topk_dense_large(
/* /*
if (idx < RADIX_SIZE) { if (idx < RADIX_SIZE) {
ptr_at(dstv, idx*dstv_strides_0) = known_bits; long long int xxx = 1337;
ptr_at(dstv, idx*dstv_strides_0) = smem[idx]; ptr_at(dstv, idx*dstv_strides_0) = xxx;
} }
return; return;
*/ */
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论