提交 bbe02b6d 编写于 作者: W Wiktor Adamski

Fixed memory leak

上级 fef913af
...@@ -32,8 +32,8 @@ Image *ScratchImageManager::Spawn( ...@@ -32,8 +32,8 @@ Image *ScratchImageManager::Spawn(
int image_count = static_cast<int>(reference_count_.size()); int image_count = static_cast<int>(reference_count_.size());
for (int i = 0; i < image_count; ++i) { for (int i = 0; i < image_count; ++i) {
int count = reference_count_[i]; int count = reference_count_[i];
if (count == 0 && images_.at(count)->dtype() == dt) { if (count == 0 && images_.at(i)->dtype() == dt) {
auto image_shape = images_.at(count)->shape(); auto image_shape = images_.at(i)->shape();
if (image_shape[0] >= shape[0] && image_shape[1] >= shape[1]) { if (image_shape[0] >= shape[0] && image_shape[1] >= shape[1]) {
found_image_idx = i; found_image_idx = i;
break; break;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册