傅立叶变换—FFT(cuda实现)
【摘要】 参考：http://www.cnblogs.com/zhangjxblog/p/5016986.html
无意间看到cuda解决FFT有一个cufft函数库，大体查看了有关cufft有关知识，写了一个解决一维情况的cuda代码，据调查知道cufft在解决1D,2D,3D的情况时间复杂度都为O(nlogn)，附上解决一维情况的代码，准备后面找一些详细的资料去学习一下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<LENGTH;i++) { CompData[i].x=Data[i]; CompData[i].y=0; } cufftComplex *d_fftData; cudaMalloc((void**)&d_fftData,LENGTH*sizeof(cufftComplex)); cudaMemcpy(d_fftData,CompData,LENGTH*sizeof(cufftComplex),cudaMemcpyHostToDevice); cufftHandle plan; cufftPlan1d(&plan,LENGTH,CUFFT_C2C,1); cufftExecC2C(plan,(cufftComplex*)d_fftData,(cufftComplex*)d_fftData,CUFFT_FORWARD); cudaDeviceSynchronize(); cudaMemcpy(CompData,d_fftData,LENGTH*sizeof(cufftComplex),cudaMemcpyDeviceToHost); for(i=0;i<LENGTH;i++) { if(CompData[i].x != 0) { printf("%3.1f",CompData[i].x); } if(CompData[i].y != 0 ) { printf("+%3.1fi",CompData[i].y); } printf("\n"); } cufftDestroy(plan); free(CompData); cudaFree(d_fftData); }
在Linux下运行的这段代码：
编译命令：nvcc -o fftcu FFT.cu -I /usr/local/cuda/include -L /usr/local/cuda/lib64 -lcufft
运行命令：./fftcu
注：/usr/local/cuda/include中有cufft.h头文件，/usr/local/cuda/lib64中有libcufft.so库文件
文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。
原文链接：blog.csdn.net/jacke121/article/details/67640120
【版权声明】本文为华为云社区用户转载文章，如果您发现本社区中有涉嫌抄袭的内容，欢迎发送邮件进行举报，并提供相关证据，一经查实，本社区将立刻删除涉嫌侵权内容，举报邮箱：
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论（0）