Skip to content

Complex autograd support for torch.{baddbmm, addbmm, addmm, addmv}#50632

Closed
anjali411 wants to merge 3 commits intogh/anjali411/84/basefrom
gh/anjali411/84/head
Closed

Complex autograd support for torch.{baddbmm, addbmm, addmm, addmv}#50632
anjali411 wants to merge 3 commits intogh/anjali411/84/basefrom
gh/anjali411/84/head

Conversation

@anjali411
Copy link
Copy Markdown
Contributor

@anjali411 anjali411 commented Jan 15, 2021

Stack from ghstack:

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.

Differential Revision: D25947471

@anjali411 anjali411 requested a review from albanD as a code owner January 15, 2021 23:39
anjali411 added a commit that referenced this pull request Jan 15, 2021
@anjali411 anjali411 requested a review from mruberry January 15, 2021 23:40
@anjali411 anjali411 added complex_autograd module: complex Related to complex number support in PyTorch labels Jan 15, 2021
…, addmv, addr}"


I'll port the following method tests in follow-up PRs:
`'baddbmm', 'addbmm', 'addmv', 'addr'`
After the tests are ported to method_tests, it would also be much easier to add tests with complex alpha and beta values.

[ghstack-poisoned]
…, addmv, addr}"


I'll port the following method tests in follow-up PRs:
`'baddbmm', 'addbmm', 'addmv', 'addr'`
After the tests are ported to method_tests, it would also be much easier to add tests with complex alpha and beta values.

[ghstack-poisoned]
@anjali411 anjali411 changed the title Complex autograd support for torch.{baddbmm, addbmm, addmm, addmv, addr} Complex autograd support for torch.{baddbmm, addbmm, addmm, addmv} Jan 17, 2021
@codecov
Copy link
Copy Markdown

codecov Bot commented Jan 17, 2021

Codecov Report

Merging #50632 (5eae9e9) into gh/anjali411/84/base (7e05d07) will decrease coverage by 0.00%.
The diff coverage is 100.00%.

@@                   Coverage Diff                    @@
##           gh/anjali411/84/base   #50632      +/-   ##
========================================================
- Coverage                 80.66%   80.65%   -0.01%     
========================================================
  Files                      1912     1912              
  Lines                    208082   208086       +4     
========================================================
- Hits                     167845   167841       -4     
- Misses                    40237    40245       +8     

Comment thread test/test_autograd.py
Copy link
Copy Markdown
Collaborator

@mruberry mruberry left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Cool!

@mruberry
Copy link
Copy Markdown
Collaborator

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.

That'd be great. The addr OpInfo looks good (see comments on PR).

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.

This makes sense. Let's just skip those tests for now with a pointer to a new issue about how those tests don't work with broadcasting.

@facebook-github-bot
Copy link
Copy Markdown
Contributor

@anjali411 merged this pull request in 227acc2.

@facebook-github-bot facebook-github-bot deleted the gh/anjali411/84/head branch January 22, 2021 15:18
laurentdupin pushed a commit to laurentdupin/pytorch that referenced this pull request Apr 24, 2026
…ytorch#50632)

Summary:
Pull Request resolved: pytorch#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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cla signed complex_autograd Merged module: complex Related to complex number support in PyTorch

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants