pytorch
227acc2e - Complex autograd support for torch.{baddbmm, addbmm, addmm, addmv} (#50632)

Commit
3 years ago
Complex autograd support for torch.{baddbmm, addbmm, addmm, addmv} (#50632) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50632 I'll port the following method tests in follow-up PRs: `'baddbmm', 'addbmm', 'addmv', 'addr'` After the tests are ported to OpInfo based tests, it would also be much easier to add tests with complex alpha and beta values. Edit- it seems like it's hard to port the broadcasting variant tests because one ends up skipping `test_inplace_grad` and `test_variant_consistency_eager` even for the case when inputs are not required to be broadcasted. Test Plan: Imported from OSS Reviewed By: navahgar Differential Revision: D25947471 Pulled By: anjali411 fbshipit-source-id: 9faa7f1fd55a1269bad282adac2b39d19bfa4591
Author
Parents
Loading