一种用于GPU异构集群的高性能FFT方法-北京大学高能效计算与应用.PDFVIP

(19)中华人民共和国国家知识产权局 (12)发明专利 (10)授权公告号 CN 105224506 B (45)授权公告日 2018.02.13 (21)申请号 201510725866.3 (22)申请日 2015.10.29 (65)同一申请的已公布的文献号 申请公布号 CN 105224506 A (43)申请公布日 2016.01.06 (73)专利权人 北京大学 地址 100871 北京市海淀区颐和园路5号 (72)发明人 陈一峯　崔翔　王韬　严磊　 (74)专利代理机构 北京万象新悦知识产权代理 事务所(普通合伙) 11360 代理人 张肖琪 (51)Int .C l. G06F 17/14(2006.01) 审查员 葛晓倩 权利要求书1页 说明书8页 附图2页 (54)发明名称 一种用于GPU异构集群的高性能FFT方法 (57)摘要 本发明公布了一种用于GPU异构集群的高性 能FFT方法，将数据进行维度划分并表示为数组 数据，按照划分维度进行分割后分布到GPU异构 集群中与划分维度相同数 目的节点上 ，再进行 FFT计算，包括：获得GPU集群的节点数 目、节点上 GPU数 目、节点内存容量、GPU设备存储器容量信 息；将数据采用数组维度表示为数组数据；针对x 维度进行维度划分；按照x维度分割数据，分布在 多个节点上 ；将y和z维度方向的数据进行FFT计 算；进行集群节点之间的数据转置；进行x维度的 FFT计算。本发明是针对GPU异构集群的多层存储 器体系结构特征的FFT方法，具有良好的性能和 B 可延展性。 6 0 5 4 2 2 5 0 1 N C CN 105224506 B 权　利　要　求　书 1/1页 1.一种用于GPU异构集群的高性能FFT方法，所述方法将待处理数据进行维度划分并表 示为数组数据，对数组数据按照划分维度进行分割后将数据分布到GPU异构集群中与划分 维度相同数目的节点上，再进行FFT计算；具体包括如下步骤： 1)测试获得所使用GPU集群的体系结构特征，包括节点数目、节点上GPU数目、节点内存 容量、GPU设备存储器容量信息；设定d为GPU异构集群的节点数目； 2)将待处理数据采用数组维度表示方法表示为数组数据；针对x维度进行维度划分，x 维度的大小设为a，记为 其中，x、y和z为维度的名称，a、b和c为相应维度的大小； 3)将步骤2)所述数组数据按照x维度分割为d份，数据分布在a个节点上；每一个节点的 数据为(a/d)*b*c个复数，此时数据记为 其中，x维度分裂为 和x 两个维度， 0 维度大小分别为d和a/d； 4)每个节点上的GPU将y维度方向和z维度方向的数据进行FFT计算；具体是进行x 维度 0 个FFT计算，即进行a/d个2维的FFT计算； 5)在进行x维度的FFT计算之前，进行集群节点之间的数据转置，将y维度的数据做进一 步的划分，得到 在集群上直接进行通信，将y维度和x维度进行交换，数 据分布变换为 得到 数据按照y 维度分布在集群节点