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

Added table index sets, and used them to make fft types enforce power… #876

Merged
merged 4 commits into from
May 3, 2022

Conversation

duvenaud
Copy link
Contributor

@duvenaud duvenaud commented Apr 19, 2022

…s-of-2 sizes.

This was very satisfying. Now we can programmatically iterate over tables of arbitrary dimension, which let me clean up the CTC and FFT examples.

The FFT inner loop also now avoids some unnecessary discrete math. The FFT code now typechecks and passes tests again, thanks to #877!

There are further enhancements that can be made to the FFT example once we can handle incrementing and decrementing Nats at the type level.

@duvenaud duvenaud changed the title [WIP] Added table index sets, and used them to make fft types enforce power… Added table index sets, and used them to make fft types enforce power… Apr 20, 2022
Copy link
Collaborator

@apaszke apaszke left a comment

Choose a reason for hiding this comment

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

Looks great! I just sent a PR that should unblock you

lib/prelude.dx Show resolved Hide resolved
@duvenaud
Copy link
Contributor Author

Okay, great, your PR unblocked me, and the FFT was already passing tests. I added a few comments and added FFT back to the tests. I think it's ready to merge now.

@apaszke apaszke merged commit 80b2617 into google-research:main May 3, 2022
@apaszke
Copy link
Collaborator

apaszke commented May 3, 2022

Thanks, David!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants