Skip to content

P
Paddle

夏有凉风，冬有雪 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

提交 00360e7e 编写于 作者: T typhoonzero
浏览文件
操作

update

上级 579c92ab
显示空白变更内容
内联 并排
Showing with 4 addition and 3 deletion
paddle/operators/lookup_table_op.cu
浏览文件 @ 00360e7e
...@@ -74,8 +74,9 @@ class LookupTableCUDAKernel : public framework::OpKernel<T> { ...@@ -74,8 +74,9 @@ class LookupTableCUDAKernel : public framework::OpKernel<T> {
dim3 threads(128, 8); dim3 threads(128, 8);
dim3 grids(8, 1); dim3 grids(8, 1);
LookupTable<T, 128, 8, LookupTable<
8><<<grids, threads, 0, context.device_context().stream()>>>( T, 128, 8,
8><<<grids, threads, 0, context.cuda_device_context().stream()>>>(
output, table, ids, N, K, D); output, table, ids, N, K, D);
} }
}; };
...@@ -135,7 +136,7 @@ class LookupTableGradCUDAKernel : public framework::OpKernel<T> { ...@@ -135,7 +136,7 @@ class LookupTableGradCUDAKernel : public framework::OpKernel<T> {
dim3 grids(8, 1); dim3 grids(8, 1);
LookupTableGrad< LookupTableGrad<
T, 128, 8, T, 128, 8,
8><<<grids, threads, 0, context.device_context().stream()>>>( 8><<<grids, threads, 0, context.cuda_device_context().stream()>>>(
d_table, d_output, ids, N, K, D); d_table, d_output, ids, N, K, D);
} }
} }
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或