pytorch

pytorch copied to clipboard

Move vmap's OrderedDict pytree support to torch.utils._pytree

1

Stack from [ghstack](https://github.com/ezyang/ghstack): * #83079 * __->__ #83073 There's no reason why it should just apply when you import vmap Test Plan: - added a new test

zou3519

[Shape Fns] Fix optional None for the actual shape functions

1

I think the person who edited `mean.dim` edited the python and the associated Torchscript version manually in two different ways. This diff fixes that all up. It also fixes the...

Gamrix

### 🐛 Describe the bug Sometimes, _post_backward_hook will not fire if gradients were not accumulated on the FSDP managed parameter, such as if all parameters in an FSDP module were...

rohan-varma

Reland "[functorch] add error inputs check to vmap test"

1

Stack from [ghstack](https://github.com/ezyang/ghstack): * #83119 * #83114 * #83077 * __->__ #83106 * #83105 * #83080 Reland of #83017. It was reverted because it was stacked on top of a...

zou3519

Fix error_inputs for linalg.lstsq; assert SampleInput args are tuple, take2

1

Stack from [ghstack](https://github.com/ezyang/ghstack): * #83119 * #83114 * #83077 * #83106 * __->__ #83105 * #83080 The previous merge attempt failed some trunk tests. Test Plan: - wait for tests

zou3519

Editing gen_jit_shape_functions to make lintrunner happy

2

Stack from [ghstack](https://github.com/ezyang/ghstack): * __->__ #79571 * #83092 * #82850 Somehow even with clang-format off, it was unhappy with this line >>> Lint for torch/csrc/jit/runtime/serialized_shape_function_registry.cpp: Warning (CLANGFORMAT) format See https://clang.llvm.org/docs/ClangFormat.html....

Gamrix

Expose an out-of-place _reduce_oop from ProcessGroupNCCL

6

Summary: This change exposes an out-of-place `_reduce_oop` API from the ProcessGroupNCCL. It allows reducing an input tensor and placing the output in a separate output tensor. Custom collectives may be...

aashaka

Update state_dict docs

2

- Better definition of `state_dict` - Additional shallow copy warning

Atcold

### 🐛 Describe the bug ```python import torch import torch.fx tracer = torch.fx.Tracer() import copy tracer_copy = copy.deepcopy(tracer) ``` ``` --------------------------------------------------------------------------- TypeError Traceback (most recent call last) [](https://localhost:8080/#) in ()...

jamesr66a

[primTorch] added ref for nn.functional.glu

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #82819 * __->__ #82214 * #82817

fdrocha