Skip to content

T
tic
Commit 56ea6d6b by Eric Junyuan Xie Committed by Tianqi Chen
Options

fix (#111)

parent 1411a5ac
Showing with 1 additions and 1 deletions
nnvm/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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment