From a45bad14c00d6949e3881475edbdebc0c28434c5 Mon Sep 17 00:00:00 2001
From: wilfChen <chenweifeng720@huawei.com>
Date: Sat, 21 Nov 2020 17:24:18 +0800
Subject: [PATCH] assign size when malloc memory

---
 mindspore/ccsrc/runtime/device/memory_manager.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mindspore/ccsrc/runtime/device/memory_manager.cc b/mindspore/ccsrc/runtime/device/memory_manager.cc
index 31ec68a827..4e122085ce 100644
--- a/mindspore/ccsrc/runtime/device/memory_manager.cc
+++ b/mindspore/ccsrc/runtime/device/memory_manager.cc
@@ -220,6 +220,7 @@ bool MemoryManager::MallocMemFromMemPool(const DeviceAddressPtr address, size_t
     return false;
   }
   address->ptr_ = device_ptr;
+  address->size_ = size;
   address->from_mem_pool_ = true;
   return true;
 }