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;