diff --git a/paddle/phi/kernels/xpu/reduce.h b/paddle/phi/kernels/xpu/reduce.h index a9ba6c1ac1347ac459ed9dbfe8fd53e3843d0b82..1507024265e23d68dadd77ae52c78f429bad6b74 100644 --- a/paddle/phi/kernels/xpu/reduce.h +++ b/paddle/phi/kernels/xpu/reduce.h @@ -78,7 +78,7 @@ int XPUReduce(const Context& dev_ctx, r = xpu::copy(dev_ctx.x_context(), reinterpret_cast(x_data), reinterpret_cast(y_data), - x.numel() * sizeof(T)); + x.numel()); PADDLE_ENFORCE_XDNN_SUCCESS(r, "copy"); } else { r = func(dev_ctx.x_context(), x_data, y_data, xdims, reduce_dims); @@ -100,7 +100,7 @@ void ReduceKernelImpl(const DeviceContext& dev_ctx, int r = xpu::copy(dev_ctx.x_context(), reinterpret_cast(x_data), reinterpret_cast(y_data), - input.numel() * sizeof(T)); + input.numel()); PADDLE_ENFORCE_XDNN_SUCCESS(r, "copy"); } else { Functor func;