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

FFT on Host and Device #18

Merged
merged 5 commits into from
Jan 5, 2024
Merged

FFT on Host and Device #18

merged 5 commits into from
Jan 5, 2024

Conversation

yasahi-hpc
Copy link
Collaborator

@yasahi-hpc yasahi-hpc commented Jan 5, 2024

In this PR, I have added the capability to call KokkosFFT APIs from both host and device. This is not quite useful so I will make it optional. (-DKokkosFFT_ENABLE_HOST_AND_DEVICE option)
If you enable this option for CUDA backend, you need fftw in addition to cufft.

Now the following operations are possible.

using execution_space = Kokkos::DefaultExecutionSpace;
using host_execution_space = Kokkos::DefaultHostExecutionSpace;
KokkosFFT::fft(execution_space(), a, out); // Calling FFTs on Device
KokkosFFT::fft(host_execution_space(), h_a, h_out); // Calling FFTs on Host

@yasahi-hpc yasahi-hpc merged commit 66b979a into main Jan 5, 2024
2 checks passed
@yasahi-hpc yasahi-hpc deleted the fft-on-host-and-device branch January 5, 2024 15:16
@yasahi-hpc yasahi-hpc mentioned this pull request Jan 5, 2024
31 tasks
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.

1 participant