HPC如何走向平民化

 

USIMD和HPCRunner开源项目Leader，目前主要聚焦于openEuler HPC SIG运营，openEuler HPC SIG致力于建立气象、分子动力学、生物和制造等领域的生态交流圈，打造HPC多样性算力部署调优统一平台，自动容器化助力极简部署，一站式调优HPC应用，让看似阳春白雪的HPC应用走向平民化！

1. 前言

说起高性能计算（High Performance Computing，缩写 HPC），很多人第一个想法就是动辄亿亿次浮点运算的“超级计算机”，国家花这么大代价做出来的机器干啥用呢，其实大到天气预报、飞机制造、新冠疫苗研发，小到唐氏筛查、《熊出没》渲染，到处可见HPC的身影，就拿最近的北京冬奥会来说，其成功举办就脱离不了济南超算对空气质量预报准确及时的预警，当然HPC也可以离我们很远，举几个戈登贝尔奖的栗子(被誉为“超级计算应用领域”内的诺贝尔奖)。

获奖年份	获奖项目	超算系统
2021	超大规模量子随机电路实时模拟	中国.神威
2020	超过一亿个原子的分子动力学模拟计算	美国.Summit
2016	千万核可扩展全球大气动力学全隐式模拟	中国.神威

对超算来说硬件很重要，但现在是“软件吞噬一切”的时代，路修的再好没有车在上面跑也没用，如果把开源软件比作汽车市场，那么HPC应用就是这个市场中的 “豪车”，为了追求最极致的性能，它们用着最先进的“发动机”(编译器)、最复杂的“轮胎”(并行技术)，同时也拥有最繁琐的“机械结构”(安装和调优流程)，优化过HPC应用的“赛车手”都有一个感受：抱黄连敲门—苦到家了，难怪HPC被喻为是IT行业“金字塔上的明珠”，本文旨在通过部署调优一个量子化学的HPC应用让大家直观的感受HPC的魅力，让看似阳春白雪的HPC也能走向平民化，走向每个人、每个家庭、每个组织。

2. 求解薛定谔方程

波函数被证明对获取材料的宏观性能(如光学性质、力学性质、热力学性质、磁学性质)极具价值，现实中的物质由多个电子构成的，它们的波函数该如何求解？一个铁原子有26个电子，两个有52个电子，计算2个铁原子相互作用时，需要计算任意一个与其它51个电子的相互作用，当计算原子数增加时，计算量呈指数增加，100个铁原子的计算量恐怕都是天文数字了，更不要说几十万个原子（纳米材料），这就不得不提著名的Kohn-Sham公式，他将多电子体系近似用一个或几个函数来代替求解多体薛定谔方程，计算量大大减少，Kohn因为这个公式获1998年Nobel化学奖，本文要介绍的HPC应用QE就是这个公式的开源实现，在量子化学领域被广泛应用，从官网介绍来看，“QUANTUM ESPRESSO是一种用于电子结构计算和材料建模的量子化学方法的软件套件，在GNU通用公共许可证下免费分发。它基于密度泛函理论，平面波基组和赝势（包括范数守恒和超软）理论。”，这段文字看不懂没关系，只要记住求解公式是这个就行了：

▲ Kohn-Sham公式

3. 安装QE

准备工作：CentOS7/openEuler+ARM服务器

QE的安装可以参考文档

(https://support.huaweicloud.com/prtg-kunpenghpcs/openmind_kunpengqe_02_0001.html)一步步安装。

但是本文推荐另一种更简便的安装方式，那就是HPC贾维斯助手(https://gitee.com/openeuler/hpcrunner)，

专门为HPC应用打造的一键依赖安装、环境配置、编译、运行、调优平台，将HPCRunner解压或者clone到服务器某个位置，其目录结构及说明如下所示：

▲ 贾维斯源码目录

步骤1：进入这个贾维斯目录并初始化环境

cd hpcunner

source ./init.sh

步骤2：拷贝QE的HPC配置并切换到改配置(基于鲲鹏GCC+OpenMPI技术栈)

cp ./templates/qe/6.4/data.qe.test.config ./

./jarvis -use data.qe.test.config

步骤3：获取QE及依赖软件

l 下载QE 6.4版本(https://github.com/QEF/q-e/archive/refs/tags/qe-6.4.1.tar.gz)，解压至tmp目录

tar -xzvf ./downloads/q-e-qe-6.4.1.tar.gz -C /tmp

l 将QE的算例也拷贝至/tmp

cp -rf ./workloads/QE/qe-test /tmp

步骤4：一键下载并安装依赖

l 执行如下命令安装鲲鹏GCC+OpenMPI (请准备咖啡，5分钟左右)

./jarvis -d -dp

▲ 下载依赖并安装

步骤5：一键编译QE

./jarvis -b

▲ QE的编译

步骤6：一键运行QE

./jarvis -r

▲ QE的运行

几分钟后应该可以看到“JOB DONE”字样，经过14次SCF自洽运算我们就可以得出原子间的能量值，到这里你已经体验了一遍HPC应用的安装运行全流程了。

4. QE调优

在HPC应用运行的过程中，可以通过perf性能工具无侵入式的采集热点函数数据，而且以目录树的形式展现出来。

步骤1：运行QE

./jarvis -r

步骤2：新建窗口并进入hpcrunner目录，运行性能采集命令

./jarvis -p

从热点函数实时分布图可以看出除了MPI通信之外，GEMM矩阵运算操作占比较高，可以考虑使用openblas进行替代，同时可以考虑把编译器换成毕昇、MPI通信库换为HyperMPI。

步骤3：拷贝优化配置

cp ./templates/qe/6.4/data.qe.test.opt.config ./

步骤4：切换到优化配置

./jarvis -use data.qe.test.opt.config

步骤5：一键下载并安装依赖(大概6分钟左右)

./jarvis -d -dp

步骤6：一键编译QE

./jarvis -b

步骤7：一键运行QE

./jarvis -r

▲ 优化之后的QE

这次的运行时间是不是更短了呢？让我们对比一下前后的性能情况吧

▲ QE前后性能对比

如果你也得到了这个结果，那么恭喜你完成了一次HPC应用的调优！

5. 总结

雄关漫道真如铁,而今迈步从头越，实际HPC应用的优化之路险阻而又漫长，越是交叉学科越需要全人类的智慧共同参与才能结出美丽的果实，但是我坚信只要一直走，总有一天会到达目的地。　　

      审核编辑：彭静