Fixed make no_optimization runs indefinitely when run the memcost exa…

…mple provided (dmlc#126) * fixed make no_optimization runs indefinitely when run the memcost example provided * format plan_memory.cc
tony32769 · Jul 12, 2017 · 1eea56e · 1eea56e
1 parent c96dd0e
commit 1eea56e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/pass/plan_memory.cc b/src/pass/plan_memory.cc
@@ -323,6 +323,10 @@ Graph PlanMemory(Graph ret) {
       ret.attrs["storage_num_not_allocated"] = std::make_shared<any>(storage_num_not_allocated);
       min_allocated_bytes = storage_allocated_bytes;
     }
+
+    if (max_match_range == 0) {
+      break;
+    }
   }
   return ret;
 }