Dzmitry Huba 1b397420f2 Enable more DTensor tests in local tensor mode and fix more integration issues (#165716) ...

- During op dispatch local tensor is supposed to collect rng state from CPU and CUDA
devices so that it can be reset before execution of the op for each such that ops
with randomness produces the same result for all ranks (note that we are planning a
separate change to add support of per rank rng state). Previously we relied on
op input arguments to deduce which devices to get rng state from. Which doesn't work
for factory functions such torch.randn. Hence this changes switches to uncondionally
collecting rng state from all devices.

- Fixing per rank specific computations in _MaskedPartial and Shard placements discovered
during test enablement.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165716
Approved by: https://github.com/ezyang

2025-10-17 23:28:22 +00:00

..

_internal

Enable more DTensor tests in local tensor mode and fix more integration issues (#165716)

2025-10-17 23:28:22 +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

Fix code descriptions in the test package. (#148145)

2025-03-04 19:14:41 +00:00