lupesko commented on a change in pull request #12746: Add env flag to disable MKLDNN cache (MXNET_MKLDNN_CACHE_ENABLED) URL: https://github.com/apache/incubator-mxnet/pull/12746#discussion_r223250487
########## File path: docs/faq/env_var.md ########## @@ -181,6 +181,10 @@ When USE_PROFILER is enabled in Makefile or CMake, the following environments ca - Values: 0, 1 ```(default=1)``` - Flag to enable or disable MKLDNN accelerator. On by default. - Only applies to mxnet that has been compiled with MKLDNN (```pip install mxnet-mkl``` or built from source with ```USE_MKLDNN=1```) + +* MXNET_MKLDNN_CACHE_ENABLED + - Values: 0, 1 ```(default=1)``` + - Flag to enable or disable MKLDNN cache. On by default. This should be disabled your dataset has variable size inputs. Review comment: I suggest we document something a bit more verbose: "Flag to enable or disable MKLDNN cache. It is on by default to optimize for performance. Should be disabled if your model has variable input shapes, as cache size may grow unbounded." ---------------------------------------------------------------- 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
