| 注册
首页|期刊导航|水科学进展|适配国产多节点GPU的二维非结构高性能水动力模型

适配国产多节点GPU的二维非结构高性能水动力模型

张大伟 冯新政 马利平 王帆 王玮琦 孙东亚

水科学进展2026,Vol.37Issue(5):917-925,9.
✕
水科学进展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

张大伟 1冯新政 1马利平 2王帆 1王玮琦 1孙东亚1

作者信息

  • 1. 中国水利水电科学研究院流域水循环与水安全全国重点实验室,北京 100038
  • 2. 西安理工大学水利水电学院,陕西 西安 710048
  • 折叠

摘要

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)

水科学进展

1001-6791

访问量0
|
下载量0
段落导航相关论文