CODEWITHSUNDEEP
pythoncudaopenclfft
PhilMacKay
PhilMacKay

Reputation: 915

Multidimensional FFT in python with CUDA or OpenCL

I have been browsing around for simple ways to program FFTs to work on my graphic card (Which is a recent NVIDIA supporting CUDA 3.something). My current option is either to learn C, then that special C version for CUDA, or use some python CUDA functions. I'd rather not learn C yet, since I only programmed in high-level languages. I looked at pyCUDA and other ways to use my graphic card in python, but I couldn't find any FFT library which could be use with python code only. Some libraries/project seem to tackle similar project (CUDAmat, Theano), but sadly I found no FFTs.

Does a function exist which could do the same thing as numpy.fft.fft2(), using my graphic card?

EDIT: Bonus point for an open source solution.

Upvotes: 4

Views: 2306

Answers (2)

fjarri
fjarri

Reputation: 9726

There's PyFFT, which is open-source and based on Apple's (somewhat limited) implementation. Disclaimer: I work on PyFFT :)

Upvotes: 4

arrayfire
arrayfire

Reputation: 1754

Yes, ArrayFire has a 2-D FFT for Python.

Disclaimer: I work on ArrayFire.

Upvotes: 3

Related Questions