albanD 6791b0c09e Change default torch_function behavior to be disabled when torch_dispatch is defined (take 2) (#120632) ... This does not introduce a new test but is tested by checking that all the classes we already have still behave as before now that they don't explicitly disable torch_function. Pull Request resolved: https://github.com/pytorch/pytorch/pull/120632 Approved by: https://github.com/ezyang		2024-03-09 01:08:37 +00:00
..
_composable	[FSDP2] Zeroed padded tensor in _apply (#121509)	2024-03-08 22:31:19 +00:00
_shard	[dist][sharded_tensor] Fix ChunkShardingSpec metadata offsets for empty shards (#121002)	2024-03-02 08:58:48 +00:00
_sharded_tensor
_sharding_spec
_spmd	Fix ouput typos (#120870)	2024-02-29 08:29:14 +00:00
_tensor	Change default torch_function behavior to be disabled when torch_dispatch is defined (take 2) (#120632)	2024-03-09 01:08:37 +00:00
_tools
algorithms
autograd
benchmarks
checkpoint	[DCP] Replaced storage() with untyped_storage() (#121538)	2024-03-08 23:46:17 +00:00
elastic	[Torch Elastic][Draft] Refactor SubprocessHandler to separate module for easier subclass (#120373)	2024-03-08 01:37:34 +00:00
examples
fsdp	[FSDP][StateDict] Allow FULL_STATE_DICT option for 2D (#120837)	2024-03-05 10:03:44 +00:00
launcher	[Torchelastic][Logging] Pluggable logsspecs using python entrypoints and option to specify one by name. (#120942)	2024-03-02 08:07:52 +00:00
nn	Fix a bug where nn.functional._AllGather.backward produces wrong gradients (#120582)	2024-02-26 09:58:27 +00:00
optim
pipeline	[c10d] Deprecate torch.distributed.pipeline (#121464)	2024-03-08 19:55:02 +00:00
rpc
tensor	[tp] remove deprecated tp_mesh_dim arg (#121432)	2024-03-08 17:46:44 +00:00
__init__.py
_composable_state.py
_functional_collectives.py	Change default torch_function behavior to be disabled when torch_dispatch is defined (take 2) (#120632)	2024-03-09 01:08:37 +00:00
_functional_collectives_impl.py	[functional collecitve] don't import torchdynamo when running torchdeploy (#120900)	2024-02-29 19:20:54 +00:00
_state_dict_utils.py	[DCP][state_dict] Let _offload_state_dict_to_cpu to return the companion_obj if it exist. (#121273)	2024-03-08 00:24:29 +00:00
argparse_util.py
c10d_logger.py
collective_utils.py
constants.py
CONTRIBUTING.md
device_mesh.py	[DeviceMesh] Ensure mesh tensor is a cpu tensor (#120046)	2024-02-22 22:03:13 +00:00
distributed_c10d.py	[c10d] Add complex support for P2P (#121240)	2024-03-08 22:47:49 +00:00
launch.py
logging_handlers.py
remote_device.py
rendezvous.py
run.py	[Torchelastic][Logging] Pluggable logsspecs using python entrypoints and option to specify one by name. (#120942)	2024-03-02 08:07:52 +00:00
utils.py