pytorch

8b3e01cd - [DTensor] implement dist_cat as a sharding prop rule (#92677)

[DTensor] implement dist_cat as a sharding prop rule (#92677) Pull Request resolved: https://github.com/pytorch/pytorch/pull/92677 Approved by: https://github.com/wanchaol