diff --git a/paddle/fluid/operators/optimizers/adam_op.cu b/paddle/fluid/operators/optimizers/adam_op.cu index 5c30962cca82594df3d91e29f6a343715d404f6b..57231e1135a6a922800810ed1515dc79c316e176 100644 --- a/paddle/fluid/operators/optimizers/adam_op.cu +++ b/paddle/fluid/operators/optimizers/adam_op.cu @@ -198,11 +198,11 @@ class AdamOpCUDAKernel : public framework::OpKernel { *mom2, ctx.GetPlace(), ctx.template device_context(), mom2_out); framework::TensorCopy( - *beta1_pow, ctx.GetPlace(), + *beta1_pow, beta1_pow->place(), ctx.template device_context(), beta1_pow_out); framework::TensorCopy( - *beta2_pow, ctx.GetPlace(), + *beta2_pow, beta2_pow->place(), ctx.template device_context(), beta2_pow_out); return; diff --git a/paddle/fluid/operators/optimizers/adam_op_npu.cc b/paddle/fluid/operators/optimizers/adam_op_npu.cc index 1169bc12ac230c0601ebe502a7eb6c3866f25381..91bb9d48bc786987a2ebac89a2eacfa5d668d30c 100644 --- a/paddle/fluid/operators/optimizers/adam_op_npu.cc +++ b/paddle/fluid/operators/optimizers/adam_op_npu.cc @@ -84,11 +84,11 @@ class AdamNPUKernel : public framework::OpKernel { *mom2, ctx.GetPlace(), ctx.template device_context(), mom2_out); framework::TensorCopy( - *beta1_pow, ctx.GetPlace(), + *beta1_pow, beta1_pow->place(), ctx.template device_context(), beta1_pow_out); framework::TensorCopy( - *beta2_pow, ctx.GetPlace(), + *beta2_pow, beta2_pow->place(), ctx.template device_context(), beta2_pow_out); return; diff --git a/paddle/fluid/operators/optimizers/adam_op_xpu.cc b/paddle/fluid/operators/optimizers/adam_op_xpu.cc index 318a8f1493010a5543c87d15d97496a1ada7f719..cb06b06824be15e74e991306188858c44472bef9 100644 --- a/paddle/fluid/operators/optimizers/adam_op_xpu.cc +++ b/paddle/fluid/operators/optimizers/adam_op_xpu.cc @@ -86,11 +86,11 @@ class AdamOpXPUKernel : public framework::OpKernel { mom2, ctx.GetPlace(), ctx.template device_context(), &mom2_out); framework::TensorCopy( - beta1_pow, ctx.GetPlace(), + beta1_pow, beta1_pow.place(), ctx.template device_context(), beta1_pow_out); framework::TensorCopy( - beta2_pow, ctx.GetPlace(), + beta2_pow, beta2_pow.place(), ctx.template device_context(), beta2_pow_out); return;