jipolanco / PencilFFTs.jl

Fast Fourier transforms of MPI-distributed Julia arrays
https://jipolanco.github.io/PencilFFTs.jl/dev/
MIT License
77 stars 7 forks source link

Feature/add cuda #37

Open corentin-dev opened 2 years ago

corentin-dev commented 2 years ago

This is a pull request to had support for CUFFT and CUDA in PencilFFTs.

CUFFT does not accept some FFTW so I had to make a specific FFTW call to plans.

I also changed a bit some of PencilArrays in order to be able to work with CuArrays.

Many changes on how internal vectors are allocated in order to be able to allocate CuArrays.