OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
a25818cf7e
pytorch/torch/distributed
History
Dzmitry Huba a51f877287 Enable local tensor mode for another set of DTensor tests (#166105) 
Enable local tensor mode DTensor tests for the optimizers, op strategy,  matrix ops,
math ops, init ops, experimental ops, embedding ops, dynamic, convolution ops, main api.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166105
Approved by: https://github.com/ezyang
2025-10-27 23:58:24 +00:00
..
_composable Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
_local_tensor Enable local tensor mode for another set of DTensor tests (#166105) 2025-10-27 23:58:24 +00:00
_pycute Enable all PIE rules on ruff (#165814) 2025-10-18 07:36:18 +00:00
_shard Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_sharded_tensor [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
_sharding_spec
_symmetric_memory Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
_tensor [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
_tools Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
algorithms Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
autograd [remove untyped defs] batch 1 (#157011) 2025-06-30 23:54:40 +00:00
benchmarks Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
checkpoint Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
elastic Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
examples Support XPU in memory tracker (#150703) 2025-06-12 21:33:52 +00:00
fsdp Fix existing pyrefly errors on main (#166312) 2025-10-27 19:03:06 +00:00
launcher [pytorch][torchelastic] Duplicate stdout and stderr and apply custom filter in torchrun (#160712) 2025-10-23 14:22:21 +00:00
nn Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
optim Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
pipelining Export flex attention with kwargs and DTensor (#166045) 2025-10-27 21:40:40 +00:00
rpc Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
tensor Re-re-re-re-apply "C++-accessible Placements via pybind11 (#163030)" (#166132) 2025-10-27 21:19:32 +00:00
__init__.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
_checkpointable.py [BE]: Backport runtime_checkable perf improvements/behavior from 3.12 (#155130) 2025-06-06 13:28:05 +00:00
_composable_state.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
_dist2.py Revert "[distributed] Replace assert statements with AssertionError exceptions (#165216)" 2025-10-14 17:05:16 +00:00
_functional_collectives_impl.py Revert "[distributed] Replace assert statements with AssertionError exceptions (#165216)" 2025-10-14 17:05:16 +00:00
_functional_collectives.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_mesh_layout.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
_serialization.py distributed/serialization: support zero sized tensors (#164198) 2025-09-30 08:11:29 +00:00
_state_dict_utils.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
argparse_util.py
c10d_logger.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
collective_utils.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
constants.py Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)" 2025-09-25 13:47:46 +00:00
CONTRIBUTING.md fix torch/distributed contributing doc (#158934) 2025-07-28 17:01:05 +00:00
device_mesh.py [DeviceMesh] Implement a device mesh concatenate api for submesh and SPMD use case (#163358) 2025-10-27 07:39:21 +00:00
distributed_c10d.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
launch.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
logging_handlers.py
remote_device.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
rendezvous.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
run.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
utils.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00