Multidimensional FFT in Python with CUDA or OpenCL

Question

I looked at simple ways to program FFT to work on my graphics card (which is recent support for NVIDIA CUDA 3.something). My current option is to either learn C, or a special version of C for CUDA, or use some of the CUDA features in Python. I would prefer not to learn C yet, since I only programmed in high-level languages. I looked at pyCUDA and other ways to use my graphics card in python, but I could not find any FFT library that could only be used with python code. Some libraries / project seem to solve a similar project (CUDAmat, Theano), but, unfortunately, I did not find the FFT.

Is there any function that can do the same as numpy.fft.fft2() using my graphics card?

EDIT: Bonus Point for an Open Source Solution.

Answer 1

Here's PyFFT , which is an open-source and Apple-based (somewhat limited) implementation . Disclaimer: I am working on PyFFT :)

Answer 2

Yes, ArrayFire has two-dimensional FFT for Python .

Disclaimer: I am working on ArrayFire.