天文学计算通常具有数据量大、计算量多的特点.平方公里阵列射电望远镜(Square Kilometre Array,SKA)等装置的建设和高性能计算平台的发展促进了天文学与高性能计算之间的联系.脉冲星搜索是SKA的主要科学方向之一.本文介绍了 一种基于OpenMP多线程和Multiprocessing多进程技术加速脉冲星搜索管线的方案,提出了 一种解决负载不平衡问题的方法,并成功地将优化后的管线安装于中国SKA区域中心原型机的x86和ARM计算节点.在默奇森大视场阵列望远镜(Murchison Widefield Array,MWA)的3个脉冲星搜索样例上的测试显示,与原始单核心管线相比,优化后的管线在x86和ARM节点上分别获得10.4-12.2倍和24.5-25.8倍的加速,并行效率达到37.1%-43.6%和24.5%-26.9%.并且ARM节点比x86节点的计算快1.1-1.3倍,显示出国产计算设备在SKA数据处理方面的巨大潜力.这项应用优化成果,近期将用于加速MWA南天快速两米巡天(Southern-sky MWA Rapid Two-metre,SMART)项目的脉冲星搜寻工作.
Parallel optimization of the pulsar search pipeline