pengzhao-intel opened a new pull request #13662: Move MKLDNN debug message under MXNET_MKLDNN_DEBUG URL: https://github.com/apache/incubator-mxnet/pull/13662 ## Description ## Move the debug message (Allocate * bytes with malloc directly) under the environment variable to avoid the confusion for the end user. Before change: > (base) [patric@mlt-gpu206 image-classification]$ python benchmark_score.py --network=resnetv1-50 --batch-size=16 > INFO:root:network: resnetv1-50 > INFO:root:device: cpu(0) > /home/patric/develop/mkldnn-update/python/mxnet/module/base_module.py:67: UserWarning: Data provided by label_shapes don't match names specified by label_names ([] vs. ['softmax_label']) > warnings.warn(msg) > [17:09:25] src/operator/nn/mkldnn/mkldnn_base.cc:74: Allocate 147456 bytes with malloc directly > [17:09:25] src/operator/nn/mkldnn/mkldnn_base.cc:74: Allocate 589824 bytes with malloc directly > [17:09:25] src/operator/nn/mkldnn/mkldnn_base.cc:74: Allocate 2359296 bytes with malloc directly > [17:09:25] src/operator/nn/mkldnn/mkldnn_base.cc:74: Allocate 9437184 bytes with malloc directly > INFO:root:batch size 16, dtype float32, images/sec: 209.347207 After change: > (base) [patric@mlt-gpu206 image-classification]$ python benchmark_score.py --network=resnetv1-50 --batch-size=16 > INFO:root:network: resnetv1-50 > INFO:root:device: cpu(0) > /home/patric/develop/mkldnn-update/python/mxnet/module/base_module.py:67: UserWarning: Data provided by label_shapes don't match names specified by label_names ([] vs. ['softmax_label']) > warnings.warn(msg) > INFO:root:batch size 16, dtype float32, images/sec: 206.424535 > With the ENV setting: > base) [patric@mlt-gpu206 image-classification]$ export MXNET_MKLDNN_DEBUG=1 > (base) [patric@mlt-gpu206 image-classification]$ python benchmark_score.py --network=resnetv1-50 --batch-size=16 > INFO:root:network: resnetv1-50 > INFO:root:device: cpu(0) > /home/patric/develop/mkldnn-update/python/mxnet/module/base_module.py:67: UserWarning: Data provided by label_shapes don't match names specified by label_names ([] vs. ['softmax_label']) > warnings.warn(msg) > [17:11:51] src/operator/nn/mkldnn/mkldnn_base.cc:516: test Convolution > [17:11:51] src/operator/nn/mkldnn/mkldnn_base.cc:516: test BatchNorm > [17:11:51] src/operator/nn/mkldnn/mkldnn_base.cc:516: test Activation > [17:11:51] src/operator/nn/mkldnn/mkldnn_base.cc:516: test Pooling > [17:11:51] src/operator/nn/mkldnn/mkldnn_base.cc:516: test Convolution > [17:11:51] src/operator/nn/mkldnn/mkldnn_base.cc:516: test BatchNorm > [17:11:51] src/operator/nn/mkldnn/mkldnn_base.cc:516: test Activation > [17:11:51] src/operator/nn/mkldnn/mkldnn_base.cc:74: Allocate 147456 bytes with malloc directly > [17:11:51] src/operator/nn/mkldnn/mkldnn_base.cc:516: test Convolution > ....
---------------------------------------------------------------- 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: [email protected] With regards, Apache Git Services
