Skip to content

Commit 462e229

Browse files
authored
[XPU] fix dense tensor usage in adam (#70856)
1 parent fb034ef commit 462e229

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/phi/kernels/xpu/adam_kernel.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,8 @@ void AdamDenseKernel(
7777

7878
float* beta1_pow_ptr = nullptr;
7979
const float* beta1_const_pow_ptr = nullptr;
80+
DenseTensor xpu_beta1_pow;
8081
if (beta1_pow.place() == CPUPlace()) {
81-
DenseTensor xpu_beta1_pow;
8282
phi::Copy(dev_ctx, beta1_pow, dev_ctx.GetPlace(), false, &xpu_beta1_pow);
8383
if (xpu_beta1_pow.dtype() == DataType::FLOAT16)
8484
funcs::GetDataPointer<Context, float>(
@@ -95,8 +95,8 @@ void AdamDenseKernel(
9595

9696
float* beta2_pow_ptr = nullptr;
9797
const float* beta2_const_pow_ptr = nullptr;
98+
DenseTensor xpu_beta2_pow;
9899
if (beta2_pow.place() == CPUPlace()) {
99-
DenseTensor xpu_beta2_pow;
100100
phi::Copy(dev_ctx, beta2_pow, dev_ctx.GetPlace(), false, &xpu_beta2_pow);
101101
if (xpu_beta2_pow.dtype() == DataType::FLOAT16)
102102
funcs::GetDataPointer<Context, float>(

0 commit comments

Comments
 (0)