There is already a patch for torch.mm. You can use that in the meantime. I am using it currently and it if working fine.
torch.mm