Helmholtz Solving and Performance Optimization in Global/Regional Assimilation and Prediction System

• 作者: Jianqiang Huang Wei Xue Haodong Bian Wenxin Yan Xiaoying Wang Wenguang Chen
• 来源期刊: 清华大学学报自然科学版（英文版）
• 年，卷（期）: 2021,（3）
• 所属期刊栏目: SPECIAL ISSUE ON HIGH PERFORMANCE COMPUTING
• 页码范围: 335-346
• 页数: 12页
• 语种: 英文
• DOI: 10.26599/TST.2019.9010057

摘要

Despite efficient parallelism in the solution of physical parameterization in the Global/Regional Assimilation and Prediction System(GRAPES),the Helmholtz equation in the dynamic core,with the increase of resolution,can hardly achieve sufficient parallelism in the solving process due to a large amount of communication and irregular access.In this paper,optimizing the Helmholtz equation solution for better performance and higher efficiency has been an urgent task.An optimization scheme for the parallel solution of the Helmholtz equation is proposed in this paper.Specifically,the geometrical multigrid optimization strategy is designed by taking advantage of the data anisotropy of grid points near the pole and the isotropy of those near memory equator in the Helmholtz equation,and the Incomplete LU(ILU)decomposition preconditioner is adopted to speed up the convergence of the improved Generalized Conjugate Residual(GCR),which effectively reduces the number of iterations and the computation time.The overall solving performance of the Helmholtz equation is improved by thread-level parallelism,vectorization,and reuse of data in the cache.The experimental results show that the proposed optimization scheme can effectively eliminate the bottleneck of the Helmholtz equation as regards the solving speed.Considering the test results on a 10-node two-way server,the solution of the Helmholtz equation,compared with the original serial version,is accelerated by 100x,with one-third of iterations reduced.