search

styfenschaer / rocket-fft

Rocket-FFT makes Numba aware of numpy.fft and scipy.fft. Rocket-FFT takes its name from the PocketFFT Fast Fourier Transformation library that powers it, and Numba's goal of making your scientific Python code blazingly fast - like a rocket. 🚀
BSD 3-Clause "New" or "Revised" License
60 stars 2 forks source link

Numba.Cuda.jit implementation ? #3

Open StephenUL opened 1 year ago

StephenUL commented 1 year ago

Hi Styfenschear,

Thank you for writing this extension and making it public, I use FFT-2D daily and this greatly reduces the time taken for the calculation. Do you plan to expand this to cuda.jit for GPU implementation? Kind Regards, Step

styfenschaer commented 1 year ago

Hi @StephenUL

I am glad that you find it helpful. Unfortunately, I lack the time, experience, and hardware to develop something that is of good quanlity and works reliably on a wide range of machines. So I very likely won't be able to do this in the near future.

Best regards Styfen