zixuanweeei opened a new pull request #17959: [MKLDNN] Add LSTMP to v1.6.x URL: https://github.com/apache/incubator-mxnet/pull/17959 ## Description ## This PR aims to support projection feature for LSTM towards the CPU platform. ## Checklist ## ### Essentials ### - [x] Upgrade dnnl(mkldnn) to its v1.4 version - [x] Change some statement of `mkldnn` to `dnnl` - [x] Implement LSTMP from both native and DNNL pass - [x] Add unit test for LSTMP - [x] Add env var to control the initializing behavior of RNN - [x] Remain input dim as int type in dnnl pass - [x] Add logging message for MXNET_RNN_USE_WEIGHT_CACHE=1 ## Comments ## - This feature requires dnnl-v1.4rc porting to MXNet v1.6.x branch. Perhaps, there are several unexpected failures of unit tests. We need to have a test of CI in advance. And then switch to the formal release of dnnl-v1.4rc. - Performance data will be added later. @ciyongch @TaoLv @pengzhao-intel
---------------------------------------------------------------- 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] With regards, Apache Git Services
