Add forward AD for torch.linalg.eigh (#62163)
authorIvan Yashchuk <ivan.yashchuk@aalto.fi>
Tue, 14 Sep 2021 04:13:56 +0000 (21:13 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 14 Sep 2021 04:15:38 +0000 (21:15 -0700)
commit0aef44cb3d8a346099ab8cec40258d263d3a262a
tree06da1ee735a7f94a61411f8e676b5897ce2f1e1b
parent35c82dbf5cb7dea4eecbfb03b9948a1faa6a1c9f
Add forward AD for torch.linalg.eigh (#62163)

Summary:
This PR adds forward mode differentiation for `torch.linalg.eigh` and a few other functions required for tests to pass.

For some reason running tests for `torch.linalg.eigvalsh` and complex `torch.linalg.eigh` hangs. These tests are skipped for now.

cc ezyang albanD zou3519 gqchen pearu nikitaved soulitzer Lezcano Varal7 jianyuh mruberry heitorschueroff walterddr IvanYashchuk xwang233

Pull Request resolved: https://github.com/pytorch/pytorch/pull/62163

Reviewed By: jbschlosser

Differential Revision: D30903988

Pulled By: albanD

fbshipit-source-id: d6a74adb9e6d2f4be8ac707848ecabf06d629823
tools/autograd/derivatives.yaml
torch/csrc/autograd/FunctionsManual.cpp
torch/csrc/autograd/FunctionsManual.h
torch/testing/_internal/common_methods_invocations.py