Hi, I’ve met exactly the same issue as you described.
In my model, there are some layers which MKLDNN does not support and also some supported layers. So I convert to mkldnn in the forward function just as you did. Same runtime error occurred.
Could u pls give me some advise on how to solve it?
Any help would be greatly appreciated.