cuda一维快速傅立叶变换的实现

参考:http://www.cnblogs.com/zhangjxblog/p/5016986.html

无意间看到cuda解决FFT有一个cufft函数库,大体查看了有关cufft有关知识,写了一个解决一维情况的cuda代码,据调查知道cufft在解决1D,2D,3D的情况时间复杂度都为O(nlogn),附上解决一维情况的代码,准备后面找一些详细的资料去学习一下cuda的函数库。

#include "stdio.h"
#include "cuda_runtime.h"
#include "cufft.h"
#include "device_launch_parameters.h"


#define LENGTH 4
int main()
{

    float Data[LENGTH] = {1,2,3,4};
    cufftComplex *CompData=(cufftComplex*)malloc(LENGTH*sizeof(cufftComplex));
    int i;
    for(i=0;i

在Linux下运行的这段代码:

编译命令:nvcc -o cuffts testFFT.cpp cudaFFT.cpp -I /usr/local/cuda/include -L /usr/local/cuda/lib64 -lcufft

运行命令:./fftcu

注:/usr/local/cuda/include中有cufft.h头文件,/usr/local/cuda/lib64中有libcufft.so库文件

你可能感兴趣的:(cuda)