Skip to content

GitLab

Explore

Sign in
Register

Primary navigation

Project

J

jaxDecomp
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Environments
- Terraform modules
- Incidents

Snippets Groups Projects

Admin message

Gitlab has been updated. More info here.

c8a38544

Commit c8a38544 authored 4 months ago by Wassim KABALAN

Downloads
- Patches
- Plain Diff

type FFT implementations

parent bd87a5ba

No related branches found

No related tags found

No related merge requests found

Changes 7

Hide whitespace changes

Inline Side-by-side

Showing

jaxdecomp/_src/cudecomp/fft.py 93 additions, 99 deletions

jaxdecomp/_src/cudecomp/fft.py
jaxdecomp/_src/fft_utils.py 76 additions, 6 deletions

jaxdecomp/_src/fft_utils.py
jaxdecomp/_src/jax/fft.py 290 additions, 114 deletions

jaxdecomp/_src/jax/fft.py
jaxdecomp/_src/jax/fftfreq.py 41 additions, 15 deletions

jaxdecomp/_src/jax/fftfreq.py
jaxdecomp/_src/pencil_utils.py 99 additions, 30 deletions

jaxdecomp/_src/pencil_utils.py
jaxdecomp/_src/spmd_ops.py 88 additions, 52 deletions

jaxdecomp/_src/spmd_ops.py
jaxdecomp/fft.py 201 additions, 105 deletions

jaxdecomp/fft.py

with 888 additions and 421 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment