From e77fd2e665544e9e1a7c9f35a4b01ada225cfe08 Mon Sep 17 00:00:00 2001 From: Thunderbrook <52529258+Thunderbrook@users.noreply.github.com> Date: Fri, 3 Sep 2021 13:38:52 +0800 Subject: [PATCH] tensor formatter (#35399) --- paddle/fluid/operators/tensor_formatter.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/tensor_formatter.cc b/paddle/fluid/operators/tensor_formatter.cc index f1b64f042c..a0cda54b31 100644 --- a/paddle/fluid/operators/tensor_formatter.cc +++ b/paddle/fluid/operators/tensor_formatter.cc @@ -119,10 +119,10 @@ void TensorFormatter::FormatData(const framework::LoDTensor& print_tensor, ? print_tensor.numel() : std::min(summarize_, print_tensor.numel()); const T* data = nullptr; + framework::LoDTensor cpu_tensor; if (is_cpu_place(print_tensor.place())) { data = print_tensor.data(); } else { - framework::LoDTensor cpu_tensor; platform::CPUPlace cpu_place; TensorCopy(print_tensor, cpu_place, &cpu_tensor); #ifdef PADDLE_WITH_ASCEND_CL -- GitLab