有效

一种基于NUMA亲和性的FFT卷积算法并行实现方法及系统

王庆林、梅松竹、郝若晨、李东升、姜晶菲、赖志权、黄显栋、刘杰
中国人民解放军国防科技大学

摘要

本发明公开了一种基于NUMA亲和性的FFT卷积算法并行实现方法及系统,方法包括:对输入数据进行快速傅里叶转换,并将第一快速傅里叶转换结果存储至指定的非一致性内存访问结点上;对权重进行快速傅里叶转换,并将第二快速傅里叶转换结果存储至指定的非一致性内存访问结点上;基于第一快速傅里叶转换结果和第二快速傅里叶转换结果实现非一致性内存访问级和多核级并行复数矩阵乘,并将复数矩阵乘的结果平均分布到所有非一致性内存访问结点上;基于复数矩阵乘的结果进行快速傅里叶逆转换,得到快速傅里叶卷积算法的输出。本发明能够显著降低NUMA架构上FFT卷积计算过程中的远程内存访问开销,提升NUMA架构上的FFT卷积的性能。