CUDA functions
This is a collection of python functions written with CUDA, using cuFFT and cuBLAS libraries. The interface with Python is written using the Python C API.
These functions intend to mimic the behavior of numpy functions: fft and correlate using the power of GPU.
Included functions
- Fast Fourier transform (Complex only)
- cuda_fft (single/double precision Fourier transfom)
- cuda_ifft (single/double precision inverse Fourier transfom)
- Autocorrelation functions (Real and complex)
- cuda_acorrelate (single/double precision autocorrelation function)
Installation
-
Run setup.py script to compile only (testing)
python setup.py build_ext --inplace -
To install the module on your system use distutils setup.py as usual:
python setup.py install --user -
Now module is available on PyPI
pip install cuda_functions -
Run and check included python scripts as example
Contact info
Abel Carreras
abelcarreras83@gmail.com
Department of Materials Science and Engineering
Kyoto University
