diff --git a/paddle/phi/kernels/gpu/adam_kernel.cu b/paddle/phi/kernels/gpu/adam_kernel.cu index 0597311e219da7c1116b36f69deb844ed23b522d..c4c9ff9e06ce0be85bc4ba33a4df143fd7f416ff 100644 --- a/paddle/phi/kernels/gpu/adam_kernel.cu +++ b/paddle/phi/kernels/gpu/adam_kernel.cu @@ -253,7 +253,7 @@ void AdamDenseKernel(const Context& dev_ctx, param.numel()); if (!use_global_beta_pow) { // Update with gpu - UpdateBetaPow<<<1, 32, 0, dev_ctx.stream()>>>( + UpdateBetaPow<<<1, 1, 0, dev_ctx.stream()>>>( beta1_, beta2_, beta1_pow.data(), @@ -352,7 +352,7 @@ void MergedAdamKernel( param[idx]->numel()); if (!use_global_beta_pow) { // Update with gpu - UpdateBetaPow<<<1, 32, 0, dev_ctx.stream()>>>( + UpdateBetaPow<<<1, 1, 0, dev_ctx.stream()>>>( beta1_, beta2_, beta1_pow[idx]->data(), diff --git a/paddle/phi/kernels/gpu/adamw_kernel.cu b/paddle/phi/kernels/gpu/adamw_kernel.cu index 6994c83f53624a4126e8d2d8c35467826b6537a3..2252deb1da2ef2942f661bc3c453d9500f52eb07 100644 --- a/paddle/phi/kernels/gpu/adamw_kernel.cu +++ b/paddle/phi/kernels/gpu/adamw_kernel.cu @@ -282,7 +282,7 @@ void AdamwDenseKernel(const Context& dev_ctx, param.numel()); if (!use_global_beta_pow) { // Update with gpu - UpdateAdamWBetaPow<<<1, 32, 0, dev_ctx.stream()>>>( + UpdateAdamWBetaPow<<<1, 1, 0, dev_ctx.stream()>>>( beta1_, beta2_, beta1_pow.data(),