eric-haibin-lin opened a new pull request #8138: add storage type logging to graph executor URL: https://github.com/apache/incubator-mxnet/pull/8138 This is mentioned in the tutorial PR #7921 and should be merged before that. The log message will be printed if env_var MXNET_INFER_STORAGE_TYPE_VERBOSE_LOGGING = 1 ``` >>> os.environ['MXNET_INFER_STORAGE_TYPE_VERBOSE_LOGGING'] = "1" >>> print(os.environ['MXNET_INFER_STORAGE_TYPE_VERBOSE_LOGGING']) 1 >>> # Data in csr format ... data = mx.sym.var('data', stype='csr', shape=(32, 10000)) >>> # Weight in row_sparse format ... weight = mx.sym.var('weight', stype='row_sparse', shape=(10000, 2)) >>> bias = mx.symbol.Variable("bias", shape=(2,)) >>> dot = mx.symbol.sparse.dot(data, weight) >>> pred = mx.symbol.broadcast_add(dot, bias) >>> y = mx.symbol.Variable("label") >>> output = mx.symbol.SoftmaxOutput(data=pred, label=y, name="output") >>> executor = output.simple_bind(ctx=mx.cpu()) [00:42:45] src/executor/../common/utils.h:130: node 0 var [00:42:45] src/executor/../common/utils.h:130: node 1 var [00:42:45] src/executor/../common/utils.h:132: node 2 dot: fcompute_ex [00:42:45] src/executor/../common/utils.h:136: input 0: csr [00:42:45] src/executor/../common/utils.h:136: input 1: row_sparse [00:42:45] src/executor/../common/utils.h:141: output 2: default [00:42:45] src/executor/../common/utils.h:130: node 3 var [00:42:45] src/executor/../common/utils.h:132: node 4 broadcast_add: fcompute [00:42:45] src/executor/../common/utils.h:136: input 2: default [00:42:45] src/executor/../common/utils.h:136: input 3: default [00:42:45] src/executor/../common/utils.h:141: output 4: default [00:42:45] src/executor/../common/utils.h:130: node 5 var [00:42:45] src/executor/../common/utils.h:132: node 6 SoftmaxOutput: fcompute [00:42:45] src/executor/../common/utils.h:136: input 4: default [00:42:45] src/executor/../common/utils.h:136: input 5: default [00:42:45] src/executor/../common/utils.h:141: output 6: default [00:42:45] src/executor/../common/utils.h:132: node 7 _backward_SoftmaxOutput: fcompute [00:42:45] src/executor/../common/utils.h:136: input 5: default [00:42:45] src/executor/../common/utils.h:136: input 6: default [00:42:45] src/executor/../common/utils.h:141: output 7: default [00:42:45] src/executor/../common/utils.h:141: output 8: default [00:42:45] src/executor/../common/utils.h:132: node 8 _backward_broadcast_add: fcompute [00:42:45] src/executor/../common/utils.h:136: input 7: default [00:42:45] src/executor/../common/utils.h:141: output 9: default [00:42:45] src/executor/../common/utils.h:141: output 10: default [00:42:45] src/executor/../common/utils.h:132: node 9 _backward_dot: fcompute_ex [00:42:45] src/executor/../common/utils.h:136: input 9: default [00:42:45] src/executor/../common/utils.h:136: input 0: csr [00:42:45] src/executor/../common/utils.h:136: input 1: row_sparse [00:42:45] src/executor/../common/utils.h:141: output 11: default [00:42:45] src/executor/../common/utils.h:141: output 12: row_sparse ``` ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org
With regards, Apache Git Services
