Fixed ResetHolder() issues

PaddlePaddle · Jan 5, 2022 · 7c6b10f · 7c6b10f
1 parent de06d82
commit 7c6b10f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/pten/core/dense_tensor.cc b/paddle/pten/core/dense_tensor.cc
@@ -291,7 +291,7 @@ void DenseTensor::ResetHolder(
   if (storage_ != nullptr && storage_->data_shared()) {
     PADDLE_ENFORCE_LE(
         numel() * SizeOf(dtype()) + meta_.offset,
-        storage_->data_shared()->size(),
+        holder->size(),
         paddle::platform::errors::InvalidArgument(
             "The size of Holder is not enough to store the Tensor."));
   }