Fix Plan memory when multiple inplace option is available (#67)

eric-haibin-lin · Oct 6, 2016 · 60ccc0d · 60ccc0d
1 parent d23d47e
commit 60ccc0d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/pass/plan_memory.cc b/src/pass/plan_memory.cc
@@ -174,6 +174,7 @@ Graph PlanMemory(Graph ret) {
         uint32_t eid_out = idx.entry_id(nid, kv.second);
         uint32_t eid_in = idx.entry_id(inode.inputs[kv.first]);
         if (ref_count[eid_in] == 1 &&
+            storage[eid_out] == GraphAllocator::kBadStorageID &&
             storage[eid_in] != GraphAllocator::kBadStorageID &&
             shape_vec[eid_out].Size() == shape_vec[eid_in].Size() &&
             dtype_vec[eid_out] == dtype_vec[eid_in]) {