pytorch

02d1cf51 - [Easy] Clean up args remap for DTensor expansion (#99040)

[Easy] Clean up args remap for DTensor expansion (#99040) Pull Request resolved: https://github.com/pytorch/pytorch/pull/99040 Approved by: https://github.com/wanchaol, https://github.com/fegin