Torch.embedding fails with RuntimeError: Placeholder storage has not been allocated on MPS device!

mps
6

Yes they are both called to(device) before evaluated.
I’ve created an issue on Github detailing this.