pytorch

3405ac8a - [TP][DTensor Op] Enable Embedding op for DTensor (#96702)

[TP][DTensor Op] Enable Embedding op for DTensor (#96702) We enabled col-wise embedding for TP users. Pull Request resolved: https://github.com/pytorch/pytorch/pull/96702 Approved by: https://github.com/wanchaol