水科学进展2026,Vol.37Issue(5):917-925,9.DOI:10.14042/j.cnki.32.1309.2026.05.011
适配国产多节点GPU的二维非结构高性能水动力模型
A high-performance 2D unstructured hydrodynamic model for domestic multi-node GPU clusters
摘要
Abstract
To meet the computational demands of numerical flood simulation under China's information technology innovation initiative,this study proposes a high-performance two-dimensional unstructured hydrodynamic model designed for domestic multi-node GPU clusters.The model is based on a Godunov-type finite volume scheme and is implemented in C++using the Heterogeneous-Compute Interface for Portability(HIP)framework,enabling efficient acceleration on domestic GPUs.METIS-based domain decomposition and the message passing interface(MPI)are adopted to support collaborative computation across multiple GPUs and computing nodes,thereby improving parallel computing efficiency.Numerical tests using the classical rectangular dam-break benchmark and the Toce River case demonstrate that the model achieves satisfactory accuracy,stable computation,and reliable inter-domain communication.Field application to Hengshan Reservoir in Jiangsu Province further shows that,for a meter-scale simulation containing 15 million mesh cells,the use of 64 GPUs across 16 nodes reduced the total runtime from 3.76 h to 15 min.This result indicates a substantial improvement in computational efficiency for large-scale,high-resolution flood simulation,reducing runtime from the hourly scale on a single GPU to the minute scale on a multi-node GPU cluster.With compatibility across domestic hardware and software environments,the proposed model has strong potential for engineering applications in high-resolution flood modeling.关键词
二维水动力模型/国产GPU/多节点/非结构网格/Godunov格式Key words
2D hydrodynamic model/domestic GPU/multi-node computing/unstructured mesh/Godunov scheme分类
建筑与水利引用本文复制引用
张大伟,冯新政,马利平,王帆,王玮琦,孙东亚..适配国产多节点GPU的二维非结构高性能水动力模型[J].水科学进展,2026,37(5):917-925,9.基金项目
国家重点研发计划项目(2024YFC3214802) (2024YFC3214802)
中国水科院科研成果转化基金专项(JZ121003A0052025) The study is financially supported by the National Key R&D Program of China(No.2024YFC3214802)and Special Fund for Scientific Achievements Transformation of IWHR,China(No.JZ121003A0052025). (JZ121003A0052025)