I don’t have 0.4 to test but the code certainly copy-pastes well to the pytorch “masterish” that I have installed. (CPU only compiled with MKL on Debian.)
I’d indeed probably try to reproduce with self-compiled master.
Best regards
Thomas