Howard Huang c0395c7800 [PP] refactor (#170804) ...

- Update `_PipelineContext` to dataclass
- Reuse a helper method
Pull Request resolved: https://github.com/pytorch/pytorch/pull/170804
Approved by: https://github.com/wconstab

2025-12-22 21:25:10 +00:00

..

_composable

[15/N] Use Python 3.10 typing (#169768)

2025-12-10 02:25:51 +00:00

_local_tensor

Part 1: LocalTensor raise ValueError for empty tensor. (#170577)

2025-12-19 15:48:04 +00:00

_pycute

[19/N] Use Python 3.10 typing (#170368)

2025-12-16 04:31:39 +00:00

_shard

Revert "[xpu][test] Port distributed checkpoint test cases on Intel GPU (#168921)"

2025-12-20 17:36:56 +00:00

_sharded_tensor

…

_sharding_spec

…

_symmetric_memory

[19/N] Use Python 3.10 typing (#170368)

2025-12-16 04:31:39 +00:00

_tensor

…

_tools

[dynamo][DebugMode] make ModTracker a no-op in compiled regions (#170124)

2025-12-13 05:24:43 +00:00

algorithms

…

autograd

…

benchmarks

…

checkpoint

[Distributed] Optimize checkpoint resharding with sweep-line algorithm (#169115)

2025-12-20 20:27:19 +00:00

debug

Upgrade pyrefly version (#169633)

2025-12-12 08:33:22 +00:00

elastic

Upgrade pyrefly version (#169633)

2025-12-12 08:33:22 +00:00

examples

…

flight_recorder

[15/N] Use Python 3.10 typing (#169768)

2025-12-10 02:25:51 +00:00

fsdp

[FSDP] Fix _unshard() passing Stream instead of Event (#170525)

2025-12-17 19:25:18 +00:00

launcher

…

nn

[19/N] Use Python 3.10 typing (#170368)

2025-12-16 04:31:39 +00:00

optim

Fix typing in public torch API (#168002)

2025-12-12 15:43:26 +00:00

pipelining

[PP] refactor (#170804)

2025-12-22 21:25:10 +00:00

rpc

…

tensor

[DTensor] single-dim foreach strategy (#170631)

2025-12-22 15:20:29 +00:00

__init__.py

…

_checkpointable.py

…

_composable_state.py

[15/N] Use Python 3.10 typing (#169768)

2025-12-10 02:25:51 +00:00

_dist2.py

[15/N] Use Python 3.10 typing (#169768)

2025-12-10 02:25:51 +00:00

_functional_collectives_impl.py

[15/N] Use Python 3.10 typing (#169768)

2025-12-10 02:25:51 +00:00

_functional_collectives.py

Optimize bs=1 case for allgather on dim 1 to not split/cat (#169404)

2025-12-16 16:30:15 +00:00

_mesh_layout.py

[DeviceMesh] Fix a corner case for coalesce in cute layout and mesh slicing (#169454)

2025-12-15 23:09:52 +00:00

_serialization.py

…

_state_dict_utils.py

[15/N] Use Python 3.10 typing (#169768)

2025-12-10 02:25:51 +00:00

argparse_util.py

…

c10d_logger.py

…

collective_utils.py

Upgrade pyrefly version (#169633)

2025-12-12 08:33:22 +00:00

constants.py

…

CONTRIBUTING.md

…

device_mesh.py

[19/N] Use Python 3.10 typing (#170368)

2025-12-16 04:31:39 +00:00

distributed_c10d.py

Fix syntax for suppression comments. (#167088)

2025-12-22 02:44:50 +00:00

launch.py

…

logging_handlers.py

…

remote_device.py

…

rendezvous.py

…

run.py

Upgrade pyrefly version (#169633)

2025-12-12 08:33:22 +00:00

utils.py

…