Yi Wang 93a0a7a0bf Fix bug visualizing 1D Tensor using rich (#152871) ... Fixes https://github.com/pytorch/pytorch/issues/152848 I didn't fix the bug earlier because the example script didn't exhaustively present all combinations of 1D/2D tensor, 1D/2D mesh, and all possible sharding specs. Therefore, in this PR, I enriched the example script to cover all possible combinations. <img width="1008" alt="f" src="https://github.com/user-attachments/assets/1745a804-a004-4f98-8332-d7498453f397" /> Pull Request resolved: https://github.com/pytorch/pytorch/pull/152871 Approved by: https://github.com/wanchaol		2025-05-07 06:04:22 +00:00
..
__init__.py	[reland][dtensor] move DTensor to public namespace (#134203)	2024-09-08 17:08:40 +00:00
_comm_mode.py	[BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547)	2025-02-28 07:35:56 +00:00
_op_coverage.py	PEP585 update - torch/distributed/tensor (#145141)	2025-01-18 20:01:59 +00:00
_visualize_sharding.py	Fix bug visualizing 1D Tensor using rich (#152871)	2025-05-07 06:04:22 +00:00
comm_mode_broswer_visual.js	[reland][dtensor] move DTensor to public namespace (#134203)	2024-09-08 17:08:40 +00:00