Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Inverse Short Time Fourier Transform in ATen native #35569

Closed
wants to merge 8 commits into from

Conversation

mthrok
Copy link
Contributor

@mthrok mthrok commented Mar 27, 2020

Ported torchaudio's implementation (test, and documentation as well) to ATen.

Note

Closes #34827
Relates #3775

@mthrok mthrok requested a review from apaszke as a code owner March 27, 2020 20:44
@mthrok mthrok requested a review from vincentqb March 27, 2020 20:45
@facebook-github-bot facebook-github-bot added the oncall: jit Add this issue/PR to JIT oncall triage queue label Mar 27, 2020
@dr-ci
Copy link

dr-ci bot commented Mar 27, 2020

💊 Build failures summary and remediations

As of commit de4d2e9 (more details on the Dr. CI page):


💚 💚 Looks good so far! There are no failures yet. 💚 💚


This comment was automatically generated by Dr. CI (expand for details).Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions on the GitHub issue tracker.

See how this bot performed.

This comment has been revised 101 times.

@mthrok mthrok force-pushed the aten-isftf branch 5 times, most recently from fed4286 to 3016752 Compare April 1, 2020 18:59
@mthrok mthrok changed the title Add Inverse Short Time Fourier Transform Add Inverse Short Time Fourier Transform in ATen native Apr 1, 2020
@mthrok mthrok force-pushed the aten-isftf branch 8 times, most recently from bb7c8b3 to 0c444a1 Compare April 7, 2020 14:14
@mthrok
Copy link
Contributor Author

mthrok commented Apr 8, 2020

FYI: There is a ongoing discussion on ways to extend istft for custom padding pytorch/audio#500 pytorch/audio#427 .

Copy link
Contributor

@vincentqb vincentqb left a comment

Choose a reason for hiding this comment

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

Overall, LGTM :) see comments below

  • Batch packing/unpacking is performed in Python. ATen implementation expects 4D input tensor.

I'm assuming you did so to mirror the stft implementation? This means that the libtorch implementation will behave differently from the python implementation, right?

The initialization in ATen seems fine to me. Can you clarify if it changes something for the user in python?

aten/src/ATen/native/SpectralOps.cpp Show resolved Hide resolved
torch/_tensor_docs.py Outdated Show resolved Hide resolved
torch/_tensor_docs.py Outdated Show resolved Hide resolved
torch/functional.py Outdated Show resolved Hide resolved
@vincentqb
Copy link
Contributor

FYI: There is a ongoing discussion on ways to extend istft for custom padding pytorch/audio#500 pytorch/audio#427 .

Let's leave these changes for a follow-up PR.

test/test_torch.py Outdated Show resolved Hide resolved
@mthrok mthrok removed the request for review from apaszke April 20, 2020 18:40
Copy link
Contributor

@vincentqb vincentqb left a comment

Choose a reason for hiding this comment

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

All the tests are green!

LGTM :)

@vincentqb
Copy link
Contributor

vincentqb commented Apr 21, 2020

@mthrok -- can you import the PR so that the internal tests run too?

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.

@mthrok has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

@mthrok mthrok deleted the aten-isftf branch April 24, 2020 19:27
@facebook-github-bot
Copy link
Contributor

@mthrok merged this pull request in 5a27ec0.

seungwonpark added a commit to seungwonpark/istft-pytorch that referenced this pull request May 19, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Merged oncall: jit Add this issue/PR to JIT oncall triage queue
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants