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

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

1 Like