fix (#111)

dmlc · Apr 18, 2017 · b279286 · b279286
1 parent d3558db
commit b279286
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pass/plan_memory.cc b/src/pass/plan_memory.cc
@@ -171,7 +171,7 @@ size_t AllocMemory(const Graph& ret, const IndexedGraph& idx, StorageVector* sto
         if (ref_count[eid_in] == 1 &&
             ref_count[eid_out] != 0 &&
             storage[eid_out] == GraphAllocator::kBadStorageID &&
-            storage[eid_in] != GraphAllocator::kBadStorageID &&
+            storage[eid_in] >= 0 &&
             shape_vec[eid_out].Size() == shape_vec[eid_in].Size() &&
             dtype_vec[eid_out] == dtype_vec[eid_in]) {
           // inplace optimization