[incubator-mxnet] branch master updated: min fix of 'storage_type' index mismatch (#14560)

[ha...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

haibin pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new e01c386  min fix of 'storage_type' index mismatch (#14560)
e01c386 is described below

commit e01c386038d16f48f0de2cab6e8a055ed137457b
Author: Songshan Gong <ie...@sina.cn>
AuthorDate: Sat Jun 8 14:04:12 2019 +0800

    min fix of 'storage_type' index mismatch (#14560)
    
    Index of array 'storage_type' should be indice of 'node entries',
    not 'nodes'.
    
    sign-of-by: songshan.gong@horizon.ai
---
 src/executor/graph_executor.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/executor/graph_executor.cc b/src/executor/graph_executor.cc
index efcb582..82e591e 100644
--- 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];
     }
   }