CUDA：实现使用CUFFT计算二维卷积(附完整源码)

源代码大师

于 2024-03-05 00:21:40 发布

阅读量275

点赞数

CC 4.0 BY-SA版权

分类专栏： CUDA实战教程文章标签：深度学习人工智能 CUDA

不予转载，严禁转载，违者必纠。

本文链接：https://blog.csdn.net/it_xiangqiang/article/details/136466629

CUDA实战教程专栏收录该内容

246 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文提供了一个使用CUDA和CUFFT库完成二维卷积的代码示例，详细介绍了如何生成二维输入信号与滤波器，以及如何通过CUDA进行卷积运算。代码包含随机信号与滤波器生成、CUDA设备内存操作、2D FFT计划创建、CUDA内核函数应用和逆FFT计算等步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

CUDA：实现使用CUFFT计算二维卷积

以下是一个使用CUDA和CUFFT库实现二维卷积的示例代码。该代码首先生成一个二维输入信号和一个二维滤波器，然后使用CUDA和CUFFT进行卷积运算。

#include <stdio.h>
#include <stdlib.h>
#include <cuda_runtime.h>
#include <cufft.h>

#define SIGNAL_SIZE_X 64
#define SIGNAL_SIZE_Y 64
#define FILTER_SIZE_X 5
#define FILTER_SIZE_Y 5

// CUDA错误检查宏
#define CUDA_CHECK_ERROR(err) \
    if (err != cudaSuccess) { \
        printf("CUDA error: %s at line %d\n", cudaGetErrorString(err), __LINE__); \
        exit(-1); \
    }

// CUFFT错误检查宏
#define CUFFT_CHECK_ERROR(err) \
    if (err != CUFFT_SUCCESS) { \
        printf("CUFFT error: %d at line %d\n", err, __LINE__); \
        exit(-1); \
    }

// 生成随机二维信号
void generate_signal(float *signal, int size_x, int size_y) {
    for (int i = 0; i < size_x * size_y;