pytorch/torch/testing
Dzmitry Huba 2cd5fd1588 Enable local tensor mode on DTensor view ops test (#165596)
While enabling this test discovered lack of support for sub meshes. Added limited support
for sub meshes by properly computing rank coordinates for a given sub mesh. The implementation
follows similar approach to collectives. We infer all sub meshes for the given dimensions and
compute each rank's coordinates with respect to is sub mesh.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165596
Approved by: https://github.com/ezyang
2025-10-16 20:52:06 +00:00
..
_internal Enable local tensor mode on DTensor view ops test (#165596) 2025-10-16 20:52:06 +00:00
__init__.py Add pyrefly suppressions 2/n (#164513) 2025-10-03 02:46:13 +00:00
_comparison.py Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +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