pytorch

b383123e - [dtensor][debug] visualize_sharding only compute offset on the first rank in mesh (#121385)

[dtensor][debug] visualize_sharding only compute offset on the first rank in mesh (#121385) **Summary** avoid computing on ranks where we do not plan to visualize the DTensor. Pull Request resolved: https://github.com/pytorch/pytorch/pull/121385 Approved by: https://github.com/wanchaol