提交 7d3b2e4b 编写于 作者: Y Yibing Liu

Fix a bug in sequence_erase_op

上级 38c61053
...@@ -55,7 +55,7 @@ __global__ void SetOutput(const T* in_dat, const int in_len, ...@@ -55,7 +55,7 @@ __global__ void SetOutput(const T* in_dat, const int in_len,
const int* num_erased, T* out_dat) { const int* num_erased, T* out_dat) {
int index = blockIdx.x * blockDim.x + threadIdx.x; int index = blockIdx.x * blockDim.x + threadIdx.x;
if (index < in_len) { if (index < in_len) {
if (in_dat[index] != in_dat[index + 1]) { if (num_erased[index] == num_erased[index + 1]) {
out_dat[index - num_erased[index]] = in_dat[index]; out_dat[index - num_erased[index]] = in_dat[index];
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册