zheng-da commented on issue #10599: [MKLDNN Bug] MKLDNN eats lots of memory and then crash down. URL: https://github.com/apache/incubator-mxnet/issues/10599#issuecomment-382989367 The memory used by MKLDNN is also allocated by the MXNet memory allocator. MKLDNN NDArrays are reused in the computation graph as normal NDArrays. The difference between MKLDNN operators and the normal operators is the temp space. MKLDNN operators use it for layout conversion. The temp space isn't required by most of the normal operators. The temp space may consume huge amount of memory and the MXNet executor may maintain multiple temp space in order to execute multiple operators simultaneously. @pengzhao-intel could you investigate the issue further? I think we should reduce the unnecessary memory consumption.
---------------------------------------------------------------- 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