min fix of 'storage_type' index mismatch (apache#14560)

Index of array 'storage_type' should be indice of 'node entries', not 'nodes'. sign-of-by: [email protected]
haohuanw · Jun 23, 2019 · dda5468 · dda5468
1 parent b5a1c80
commit dda5468
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/executor/graph_executor.cc b/src/executor/graph_executor.cc
@@ -1023,8 +1023,8 @@ void GraphExecutor::InitDataEntryMemory(std::vector<NDArray>* shared_pool) {
     for (uint32_t i = 0; i < idx[nid].source->num_outputs(); ++i) {
       auto eid = idx.entry_id(nid, i);
       data_context[eid] = vctx[nid];
-      CHECK_NE(vstorage_type[nid], kUndefinedStorage);
-      data_storage_type[eid] = (NDArrayStorageType) vstorage_type[nid];
+      CHECK_NE(vstorage_type[eid], kUndefinedStorage);
+      data_storage_type[eid] = (NDArrayStorageType) vstorage_type[eid];
     }
   }