fix adam thread num (#48297)

dd27996c · sneaxiy · GitHub · bcf75132 · dd27996c · dd27996c
隐藏空白更改
内联并排

Showing with 3 addition and 3 deletion

paddle/phi/kernels/gpu/adam_kernel.cu paddle/phi/kernels/gpu/adam_kernel.cu +2 -2

paddle/phi/kernels/gpu/adamw_kernel.cu paddle/phi/kernels/gpu/adamw_kernel.cu +1 -1

未找到文件。
--- 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<MPDType><<<1, 32, 0, dev_ctx.stream()>>>(
+      UpdateBetaPow<MPDType><<<1, 1, 0, dev_ctx.stream()>>>(
          beta1_,
          beta2_,
          beta1_pow.data<MPDType>(),
@@ -352,7 +352,7 @@ void MergedAdamKernel(
          param[idx]->numel());
      if (!use_global_beta_pow) {
        // Update with gpu
-        UpdateBetaPow<MPDType><<<1, 32, 0, dev_ctx.stream()>>>(
+        UpdateBetaPow<MPDType><<<1, 1, 0, dev_ctx.stream()>>>(
            beta1_,
            beta2_,
            beta1_pow[idx]->data<MPDType>(),

--- 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<MPDType><<<1, 32, 0, dev_ctx.stream()>>>(
+      UpdateAdamWBetaPow<MPDType><<<1, 1, 0, dev_ctx.stream()>>>(
          beta1_,
          beta2_,
          beta1_pow.data<MPDType>(),