Aaron Gokaslan
2f95a3d0fc
[BE]: Apply ruff PERF fixes to torch ( #104917 )
...
Applies automated ruff fixes in the PERF modules and enables all automatic ones. I also updated ruff which applied some additional fixes.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/104917
Approved by: https://github.com/ezyang , https://github.com/albanD
2023-07-11 20:45:21 +00:00
ts
d2d03f0f44
Make index_add_ error if input source shape is wrong ( #100321 )
...
Fixes #92576 , checking the following as described in the documentation:
"source.shape[dim] == len(index) and source.shape[i] == self.shape[i] for i != dim"
Would be happy to iterate on this if there are any issues, and would be happy to implement the checking for the CUDA and MPS implementations of index_add_.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/100321
Approved by: https://github.com/lezcano
2023-06-08 06:51:10 +00:00
Ivan Yashchuk
fba13d94a1
Remove deprecated torch.symeig ( #70988 )
...
The time has come to remove deprecated linear algebra related functions. This PR removes `torch.symeig`.
- [x] XLA PR: https://github.com/pytorch/xla/pull/4498
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70988
Approved by: https://github.com/lezcano , https://github.com/kit1980 , https://github.com/malfet
2023-01-31 11:59:11 +00:00
Kshiteej K
845e4b8a47
[fix] legacybatching: getPhysicalDims ( #93261 )
...
Fixes #92985
Minimum Repro:
```python
import torch
from torch._vmap_internals import vmap
input = torch.randn(2, 2)
def fn(x):
return x.sum(())
o = vmap(fn)(input)
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/93261
Approved by: https://github.com/albanD , https://github.com/Skylion007
2023-01-30 21:06:32 +00:00
PyTorch MergeBot
acdd462b1a
Revert "Remove deprecated torch.symeig ( #70988 )"
...
This reverts commit d70ed68162https://github.com/pytorch/pytorch/pull/70988 on behalf of https://github.com/kit1980 due to Failing XLA tests, forward fix unsuccessful
2023-01-24 19:03:40 +00:00
Ivan Yashchuk
d70ed68162
Remove deprecated torch.symeig ( #70988 )
...
The time has come to remove deprecated linear algebra related functions. This PR removes `torch.symeig`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70988
Approved by: https://github.com/lezcano , https://github.com/kit1980
2023-01-23 22:51:40 +00:00
lezcano
46a81c8db7
Deprecate .mT,.T,.mH,.H on 0D tensors ( #92143 )
...
As discussed with @ngimel, this is not only not documented,
but also an unnecessary edge case. See https://github.com/pytorch/pytorch/pull/90463#discussion_r1064807197
Pull Request resolved: https://github.com/pytorch/pytorch/pull/92143
Approved by: https://github.com/ngimel
2023-01-17 16:54:35 +00:00
Richard Zou
fb2e1878cb
[torch.func] alias torch.func.vmap as torch.vmap ( #91026 )
...
This PR also redirects torch.vmap to torch.func.vmap instead of the old
vmap prototype.
Test Plan:
- tests
- view docs preview
Pull Request resolved: https://github.com/pytorch/pytorch/pull/91026
Approved by: https://github.com/albanD , https://github.com/samdow
2022-12-21 20:51:49 +00:00
Richard Zou
4b1053497c
[vmap] Prepend "legacy" to files for old vmap implementation ( #90324 )
...
We have an older torch.vmap implementation. It is no longer supported.
It still needs to exist somewhere for the sake of BC with
torch.autograd.functional.
This PR makes it clear what files are meant for implementing the old
vmap implementation. I've seen a couple of PRs recently adding support
for the old vmap implementation, so this will lessen the confusion.
Test Plan:
- CI
Pull Request resolved: https://github.com/pytorch/pytorch/pull/90324
Approved by: https://github.com/samdow
2022-12-07 18:46:15 +00:00
