Skip to content

Commit e77fd2e

Browse files
authored
tensor formatter (#35399)
1 parent ccd42db commit e77fd2e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/operators/tensor_formatter.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,10 +119,10 @@ void TensorFormatter::FormatData(const framework::LoDTensor& print_tensor,
119119
? print_tensor.numel()
120120
: std::min(summarize_, print_tensor.numel());
121121
const T* data = nullptr;
122+
framework::LoDTensor cpu_tensor;
122123
if (is_cpu_place(print_tensor.place())) {
123124
data = print_tensor.data<T>();
124125
} else {
125-
framework::LoDTensor cpu_tensor;
126126
platform::CPUPlace cpu_place;
127127
TensorCopy(print_tensor, cpu_place, &cpu_tensor);
128128
#ifdef PADDLE_WITH_ASCEND_CL

0 commit comments

Comments
 (0)