don't exit if requested_size < size (#37880)

don't exit if requested_size < size
PaddlePaddle · Dec 7, 2021 · 4035bd2 · 4035bd2
1 parent 508b756
commit 4035bd2
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/paddle/fluid/framework/tensor.cc b/paddle/fluid/framework/tensor.cc
@@ -62,14 +62,7 @@ void* Tensor::mutable_data(const platform::Place& place,
           "The Tensor's shape is [",
           dims(), "] now"));
   size_t size = numel() * SizeOfType(type);
-  if (requested_size) {
-    PADDLE_ENFORCE_GE(
-        requested_size, size,
-        platform::errors::InvalidArgument(
-            "The requested memory size is less than the memory size of Tensor. "
-            "But received requested memory size is %d, "
-            "memory size of Tensor is %d.",
-            requested_size, size));
+  if (requested_size && (requested_size > size)) {
     size = requested_size;
   }
   /* some versions of boost::variant don't have operator!= */