提交 0f91beef 编写于 作者: M minqiyang

Fix bug

test=develop
上级 3d784c27
......@@ -53,7 +53,7 @@ struct EmbeddingVSumFunctor {
PADDLE_ENFORCE_LT(ids[begin], row_number);
PADDLE_ENFORCE_GE(ids[begin], 0, "ids %d", i);
blas.VCOPY(row_width, table + ids[begin] * row_width,
blas.VCOPY(row_width, table + ids[begin + j] * row_width,
output + i * last_dim + j * row_width);
}
......@@ -62,7 +62,7 @@ struct EmbeddingVSumFunctor {
PADDLE_ENFORCE_LT(ids[r], row_number);
PADDLE_ENFORCE_GE(ids[r], 0, "ids %d", i);
blas.AXPY(row_width, 1., table + ids[r] * row_width,
output + i * row_width + (r % ids_count) * row_width);
output + i * last_dim + (r % ids_count) * row_width);
}
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册