azai91 edited a comment on issue #12746: Add env flag to disable MKLDNN cache (MXNET_MKLDNN_CACHE_ENABLED) URL: https://github.com/apache/incubator-mxnet/pull/12746#issuecomment-431525464 Resnet with cache disabled (set to 10). ubuntu@ip-172-31-3-217:~/incubator-mxnet$ MXNET_MKLDNN_CACHE_SIZE=10 python example/image-classification/benchmark_score.py [23:27:39] ../src/operator/nn/mkldnn/mkldnn_base.cc:74: Allocate 147456 bytes with malloc directly [23:27:39] ../src/operator/nn/mkldnn/mkldnn_base.cc:74: Allocate 589824 bytes with malloc directly [23:27:39] ../src/operator/nn/mkldnn/mkldnn_base.cc:74: Allocate 2359296 bytes with malloc directly [23:27:39] ../src/operator/nn/mkldnn/mkldnn_base.cc:74: Allocate 9437184 bytes with malloc directly INFO:root:batch size 32, dtype float32, images/sec: 83.920928
---------------------------------------------------------------- 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
