CPU-GPU协同计算在MOC中子输运异构并行计算中的应用研究  

Study on Heterogeneous Computing for MOC Neutron Transport Calculation with CPU-GPU Concurrent Calculation

作　　者：宋佩涛 张志俭 张乾 梁亮 赵强 Song Peitao;Zhang Zhijian;Zhang Qian;Liang Liang;Zhao Qiang(Fundamental Science on Nuclear Safety and Simulation Technology Laboratory,Harbin Engineering University,Harbin,150001,China)

机构地区：[1]哈尔滨工程大学核安全与仿真技术国防重点学科实验室,哈尔滨150001

出　　处：《核动力工程》2020年第4期17-21,共5页Nuclear Power Engineering

摘　　要：特征线方法(MOC)可以精确求解任意几何的中子输运方程,但该方法收敛慢、计算时间长。本研究基于空间区域分解和特征线并行技术,采用MPI+Open MP/CUDA编程模型,实现了适用于中央处理器-图形处理器(CPU-GPU)异构系统的二维MOC异构并行算法。为充分利用异构系统中的CPU和GPU计算资源,实现CPU-GPU协同计算,提出动态任务分配模型,根据CPU和GPU的计算能力合理分配计算任务。数值验证结果表明:程序具有良好的计算精度;动态任务分配模型能根据硬件性能给出最佳任务分配方案;5异构节点(包含20块GPU)并行时,相对MPI+CUDA并行模式,采用CPU-GPU协同计算后,程序整体效率提升达到14%。The Method of Characteristics(MOC) is capable to accurately solve the neutron transport equation with arbitrary geometry. However, the MOC suffers from some drawbacks: slow convergence and time consuming. Based on the spatial domain decomposition and the ray parallelization, the parallel 2D MOC algorithm was implemented with MPI+OepnMP/CUDA programming model to leverage the computing power of Central Processing Unit-Graphics Processing Unit(CPU-GPU) heterogeneous high-performance computing systems. In addition, a dynamic workload partitioning scheme was proposed to efficiently take advantage of all the CPU and GPU resources. The workload is appropriately assigned to the CPU and GPU according to their computational capabilities, and all CPUs and GPUs perform the calculation concurrently. The numerical results demonstrate that the parallel algorithm maintains the desired accuracy. Meanwhile, the dynamic workload portioning scheme can provide the optimal workload partition based on the runtime performance. As a result, about 14% improvement is observed in the overall performance compared with the MPI+CUDA parallelization when the CPU-GPU heterogeneous computation is performed on 5 heterogeneous nodes(including 20 GPUs).

关 键 词：异构并行 特征线方法 中子输运计算 GPU CUDA 

分 类 号：TL329[核科学技术—核技术及应用]