mailcorahul:

One is I don’t get the speedup as you get with mkldnn. Infact it is taking more time with mkldnn.


This is weird, I would make some more tests with larger and smaller tensors. If MKLDNN doesn’t provide speedup, it means your build is probably broken.



 mailcorahul:

Secondly I get the following error whenever there I try to to use mkldnn.to_mkldnn(pretrained_model) such as resnet18 or resnet50


For these architectures, I guess you could easily adapt it to support MKLDNN, just replace the view calls by reshape.