From 9cfa811e034a155df199f672a46010c06df58683 Mon Sep 17 00:00:00 2001 From: Zhanlue Yang Date: Mon, 24 Jan 2022 11:25:50 +0800 Subject: [PATCH] Fixed ResizeAndAllocate issues (#39101) --- paddle/pten/api/lib/utils/tensor_utils.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/pten/api/lib/utils/tensor_utils.cc b/paddle/pten/api/lib/utils/tensor_utils.cc index 2e94d508aec..e9f5ec2d057 100644 --- a/paddle/pten/api/lib/utils/tensor_utils.cc +++ b/paddle/pten/api/lib/utils/tensor_utils.cc @@ -311,7 +311,7 @@ void MovesStorageBase(pten::DenseTensor* src, paddle::framework::Tensor* dst) { dst, platform::errors::InvalidArgument( "The destination Tensor is nullptr when move storage.")); - dst->ResizeAndAllocate(src->dims()); + dst->Resize(src->dims()); dst->set_type(pten::TransToProtoVarType(src->dtype())); auto storage = src->MoveMemoryHolder(); dst->ResetHolderWithType(storage, pten::TransToProtoVarType(src->dtype())); @@ -332,7 +332,7 @@ void SharesStorageBase(pten::DenseTensor* src, paddle::framework::Tensor* dst) { dst, platform::errors::InvalidArgument( "The destination Tensor is nullptr when move allocation.")); - dst->ResizeAndAllocate(src->dims()); + dst->Resize(src->dims()); dst->ResetHolderWithType(src->Holder(), pten::TransToProtoVarType(src->dtype())); dst->set_offset(src->meta().offset); @@ -374,7 +374,7 @@ void MakeVariableFromPtenTensor(pten::DenseTensor* src, auto* tensor = variable->GetMutable(); auto dtype = pten::TransToProtoVarType(src->dtype()); - tensor->ResizeAndAllocate(src->dims()); + tensor->Resize(src->dims()); SetLoD(tensor->mutable_lod(), src->lod()); if (!tensor->IsInitialized() || -- GitLab