Dzmitry Huba 86f9f1d0ab Enable local tensor model for DTensor redistribute tests (#166081) ... Redistribute test exercise extensively various sharding schemes and redistribution between them. These tests uncovered more edge cases that were not supported by the local tensor primarily different flavors of uneven sharding. In order to handle these cases this change implements missing functional collectives and adds support for uneven sharding case where sharding group (ranks) is larger than the size of the dimension being sharded. In the latter case the "missing" shards are represented by zero sized tensors so that the rest of the local tensor machinery can stay oblivious to this special case. Pull Request resolved: https://github.com/pytorch/pytorch/pull/166081 Approved by: https://github.com/ezyang		2025-10-26 22:21:43 +00:00
..
_internal	Enable local tensor model for DTensor redistribute tests (#166081)	2025-10-26 22:21:43 +00:00
__init__.py	Add pyrefly suppressions 2/n (#164513)	2025-10-03 02:46:13 +00:00
_comparison.py	[1/N] Add strict parameter to Python zip calls (#165531)	2025-10-18 05:26:33 +00:00
_creation.py	[BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552)	2025-08-07 00:09:56 +00:00
_utils.py	Fix code descriptions in the test package. (#148145)	2025-03-04 19:14:41 +00:00