未验证 提交 360cb183 编写于 作者: Q QI JUN 提交者: GitHub

fix bug in lookup table grad operator (#5228)

上级 9074a60c
...@@ -116,7 +116,7 @@ class LookupTableGradCUDAKernel : public framework::OpKernel<T> { ...@@ -116,7 +116,7 @@ class LookupTableGradCUDAKernel : public framework::OpKernel<T> {
auto* d_output_data = d_output->data<T>(); auto* d_output_data = d_output->data<T>();
PADDLE_ENFORCE_EQ(d_table_value->dims(), d_output->dims()); PADDLE_ENFORCE_EQ(d_table_value->dims(), d_output->dims());
memory::Copy(gpu_place, d_table_data, gpu_place, d_output_data, memory::Copy(gpu_place, d_table_data, gpu_place, d_output_data,
d_output->numel(), stream); d_output->numel() * sizeof(T), stream);
} else { } else {
auto ids_t = context.Input<LoDTensor>("Ids"); auto ids_t = context.Input<LoDTensor>("Ids");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册