Skip to content

Conversation

@vishwakftw
Copy link
Contributor

@vishwakftw vishwakftw commented Jun 17, 2019

Changelog:

  • Port symeig from TH/THC to ATen
  • Enable batching of matrix inputs for symeig
  • Modify derivative computation based on batching
  • Update docs to reflect the change

Test Plan:

  • Added additional tests in test_torch.py (with a port to test_cuda.py) and common_methods_invocations.py to test if both the port and batching work.

@pytorchbot pytorchbot added module: cpu CPU specific problem (e.g., perf, algorithm) module: cuda Related to torch.cuda, and CUDA support in general module: internals Related to internal abstractions in c10 and ATen module: operators module: pybind Related to our Python bindings / interactions with other Python libraries labels Jun 17, 2019
@pytorchbot pytorchbot added the module: cublas Problem related to cublas support label Jun 18, 2019
@vishwakftw vishwakftw changed the title [WIP] Port symeig to ATen with batching Port symeig to ATen and enable batching of inputs Jun 19, 2019
@pytorchbot pytorchbot added module: docs Related to our documentation, both in docs/ and docblocks module: tests Issues related to tests (not the torch.testing module) labels Jun 19, 2019
@vishwakftw
Copy link
Contributor Author

@pytorchbot rebase this please

@vishwakftw
Copy link
Contributor Author

Windows build failures are spurious.

@vishwakftw
Copy link
Contributor Author

@pytorchbot rebase this please

@vishwakftw
Copy link
Contributor Author

@pytorchbot rebase this please

Copy link
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

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

@soumith is landing this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

@vishwakftw
Copy link
Contributor Author

Thank you for the review.

@vishwakftw
Copy link
Contributor Author

@soumith is there I need to do to help get this landed?

zdevito pushed a commit to zdevito/ATen that referenced this pull request Jun 25, 2019
Summary:
Changelog:
- Port `symeig` from TH/THC to ATen
- Enable batching of matrix inputs for `symeig`
- Modify derivative computation based on batching
- Update docs to reflect the change
Pull Request resolved: pytorch/pytorch#21858

Test Plan: - Added additional tests in `test_torch.py` (with a port to `test_cuda.py`) and `common_methods_invocations.py` to test if both the port and batching work.

Differential Revision: D15981789

Pulled By: soumith

fbshipit-source-id: ab9af8361f8608db42318aabc8421bd99a1ca7ae
@facebook-github-bot
Copy link
Contributor

@soumith merged this pull request in bcb5fd8.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Merged module: cpu CPU specific problem (e.g., perf, algorithm) module: cublas Problem related to cublas support module: cuda Related to torch.cuda, and CUDA support in general module: docs Related to our documentation, both in docs/ and docblocks module: internals Related to internal abstractions in c10 and ATen module: pybind Related to our Python bindings / interactions with other Python libraries module: tests Issues related to tests (not the torch.testing module) open source

Projects

None yet

Development

Successfully merging this pull request may close these issues.

6 participants