计算机工程

作者：郭渝洛, 边浩东, 董润婷, 唐嘉豪, 王晓英, 黄建强

单位：青海大学 计算机技术与应用系, 西宁 810016

关键词：冷冻电镜,负载均衡,缓存,单指令多数据流矢量化,傅里叶空间,图像相似度计算,并行计算

基金：国家自然科学基金“基于GPU异构体系结构的大规模图数据挖掘关键技术研究”（62062059）；国家自然科学基金“智能电网影响感知的数据中心需求响应策略与机制研究”（61767024）；青海省科技计划项目“GRAPES数值天气预报模式动力框架并行并行算法与优化技术研究”（2019-ZJ-7034）；国家重点实验室开放基金“基于三江源数据分析平台的全球数值天气预报系统的性能优化技术研究”（2020-ZZ-03）。

在冷冻电镜三维重建计算模型中，傅里叶空间图像相似度算法被频繁调用，大量的计算开销导致模型整体运行速度变慢。针对该问题，提出一种基于SIMD的并行傅里叶空间图像相似度算法。通过手动负载均衡方式最大化CPU的线程使用率，添加AVX-512指令集增强CPU矢量寄存器的作用，同时设计高效的数据结构提升内存访问效率。在Intel Xeon Platinum 9242平台上的实验结果表明，与原始的OpenMP并行模型相比，经该算法优化后的程序可获得平均5.132倍的加速比，并且具有较强的鲁棒性。

来源：2021年第11期

《计算机工程》期刊编辑部

查看计算机工程杂志2021年第11期