mozga-intel commented on issue #19265: URL: https://github.com/apache/incubator-mxnet/issues/19265#issuecomment-712736973
Hi @Zha0q1 There’s a bug in oneDNN LSTM forward inference that results in using ~4x more memory for LSTM workspace in inference cases. Could you please tell me whether this addressing (look at the table), is acceptable and it allows you to resolve any issues? | Before | After -- | -- | -- The total size of memory needed to allocate LSTM tensor (dim: 20756, 500) | 230 GB (~4x more memory) | 56 GB (~4x less memory) ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
