I am on master and my error is a bit different though. Also unfortunately, your solution doesn’t work for my case. 
I get this error:
RuntimeError: the derivative for embedding_dense_backward is not implemented