提交 048bc12d 编写于 作者: G guosheng

Fix GPU kernel of lookup_table_op enhancement

上级 e54eaee5
...@@ -37,7 +37,7 @@ __global__ void LookupTable(T* output, const T* table, const int64_t* ids, ...@@ -37,7 +37,7 @@ __global__ void LookupTable(T* output, const T* table, const int64_t* ids,
const T* tab = table + id * D; const T* tab = table + id * D;
for (int i = idx; i < D; i += BlockDimX) { for (int i = idx; i < D; i += BlockDimX) {
if (PaddingFlag) { if (PaddingFlag) {
if (idx == padding_idx) if (id == padding_idx)
out[i] = static_cast<T>(0); out[i] = static_cast<T>(0);
else else
out[i] = tab[i]; out[i] = tab[i];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册