I have PyTorch 1.7.1 and looking at this: https://github.com/pytorch/pytorch/blob/master/torch/utils/mkldnn.py I don’t see support for MKLDNN for bidirectional LSTMs on CPUs. Also, when I run my code in verbose, I find calls only for MKL not for MKLDNN. I checked my pytorch configuration which shows both MKL and MKLDNN are ON.
Am I missing something or there is no support of MKLDNN for bidirectional LSTMs on CPUs in Pytorch?
I am doing a comparison of LSTMs on Tensorflow/Keras and PyTorch so just want to be sure that I am not missing anything.
1.4m articles
1.4m replys
5 comments
56.9k users