From 87cf8ef7478f1e93dc3ffd27af7eadbafa206d46 Mon Sep 17 00:00:00 2001
From: wang_shaocong <wangshaocong1@huawei.com>
Date: Fri, 16 Apr 2021 11:56:05 +0800
Subject: [PATCH] [MSLITE] fix bug of memory leakage.

---
 mindspore/lite/src/runtime/allocator.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/lite/src/runtime/allocator.cc b/mindspore/lite/src/runtime/allocator.cc
index 90e91ad364..2a423d9120 100644
--- a/mindspore/lite/src/runtime/allocator.cc
+++ b/mindspore/lite/src/runtime/allocator.cc
@@ -55,7 +55,7 @@ void *DefaultAllocator::Malloc(size_t size) {
   }
   Lock();
   auto iter = freeList_.lower_bound(size);
-  if (iter != freeList_.end() && (iter->second->size >= size) && (iter->second->size < (size << shiftFactor_))) {
+  if (iter != freeList_.end() && (iter->second->size >= size) && (iter->second->size <= (size << shiftFactor_))) {
     auto membuf = iter->second;
     freeList_.erase(iter);
     allocatedList_[membuf->buf] = membuf;