JIT Tracing fails with basic EmbeddingBag when changing devices

Yes, based on this message from a code owner.

I don’t know what the desired export path will be.

EDIT: here torch.export is mentioned for the 2.0 stack, which seems to be the planned approach.

1 Like