论文标题

将微物理模型CASIM移植到GPU和KNL Cray机器

Porting the microphysics model CASIM to GPU and KNL Cray machines

论文作者

Brown, Nick, Nigay, Alexandr, Weiland, Michèle, Hill, Adrian, Shipway, Ben

论文摘要

CASIM是一种微物理方案,它计算大气中的水分液滴之间的相互作用,并构成了天气和气候建模代码的关键部分。但是,所涉及的计算是计算密集型的,因此研究Casim是否可以利用新型的硬件体系结构,并且性能的可能提高可能是有意义的。 在本文中,我们介绍了通过指令驱动的OpenACC将CASIM移植到GPU的工作,并修改Casim使用OpenMP利用骑士降落(KNL)处理器。由于设计,将特定的计算内核提取出来卸载到GPU的模型被证明是次优的,而整个方案已移植到GPU上。我们认为OpenACC对这种方法的适用性和成熟度以及已确定的重要优化。使Casim能够利用KNL非常容易,但仍然需要仔细的实验​​才能了解最佳的设计和配置。在确定有关Casim和其他类似模型对这些体系结构的适用性之前,讨论了这两个版本的CASIM的性能,与最新一代CPU相比。这项工作的结果是CASIM的版本,它在使用GPU和KNLS时显示出有希望的性能优势,并使社区能够利用这些技术,除了可以应用于其他类似的天气和气候模型的一般技术外。

CASIM is a microphysics scheme which calculates the interaction between moisture droplets in the atmosphere and forms a critical part of weather and climate modelling codes. However the calculations involved are computationally intensive and so investigating whether CASIM can take advantage of novel hardware architectures and the likely increase in performance this might afford makes sense. In this paper we present work done in porting CASIM to GPUs via the directive driven OpenACC and also modifying CASIM to take advantage of the Knights Landing (KNL) processor using OpenMP. Due to the design, models extracting out specific computational kernels for offload to the GPU proved suboptimal and instead the entire scheme was ported over to the GPU. We consider the suitability and maturity of OpenACC for this approach as well as important optimisations that were identified. Enabling CASIM to take advantage of the KNL was significantly easier, but still required careful experimentation to understand the best design and configuration. The performance of both versions of CASIM, in comparison to the latest generation of CPUs is discussed, before identifying lessons learnt about the suitability of CASIM and other similar models for these architectures. The result of this work are versions of CASIM which show promising performance benefits when utilising both GPUs and KNLs and enable the communities to take advantage of these technologies, in addition to general techniques that can be applied to other similar weather and climate models.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源