diff --git a/paddle/fluid/operators/lookup_table_v2_op.cu b/paddle/fluid/operators/lookup_table_v2_op.cu index 5c98eab403096bcc39251445145f16dc613d314e..b3b0f8f1960901226a2f4d5e59e7aac47907a5bf 100644 --- a/paddle/fluid/operators/lookup_table_v2_op.cu +++ b/paddle/fluid/operators/lookup_table_v2_op.cu @@ -105,17 +105,17 @@ class LookupTableV2CUDAKernel : public framework::OpKernel { auto *table = table_t->data(); auto *output = output_t->mutable_data(context.GetPlace()); - dim3 threads(128, 8); - dim3 grids(8, 1); + dim3 threads(256, 4); + dim3 grids(80, 1); if (padding_idx == -1) LookupTableV2< - T, 128, 8, 8, + T, 256, 4, 80, false><<>>( output, table, ids, N, K, D, padding_idx); else LookupTableV2< - T, 128, 8, 8, + T, 256, 4, 80, true><<>>( output, table, ids, N, K, D, padding_idx); }