diff --git a/paddle/fluid/operators/softmax_op_xpu.cc b/paddle/fluid/operators/softmax_op_xpu.cc
index 5d190189bf08253e8e1c04830c72b7d0abe1dfbe..ed7034ef6ab416a4e98ddcd02f045af459298d65 100644
--- a/paddle/fluid/operators/softmax_op_xpu.cc
+++ b/paddle/fluid/operators/softmax_op_xpu.cc
@@ -46,7 +46,7 @@ class SoftmaxXPUKernel : public framework::OpKernel<T> {
     Tensor clip_x;
     int len = x->numel();
     T* clip_x_data =
-        clip_x.mutable_data<T>(platform::XPUPlace(), len * sizeof(T));
+        clip_x.mutable_data<T>(context.GetPlace(), len * sizeof(T));
     r = xpu::clip(dev_ctx.x_context(), x->data<float>(), clip_x_data, len,
                   -1e30, 1e30);
     PADDLE_ENFORCE_EQ(r, XPU_SUCCESS,