[论文解读] QIRAL: A High Level Language for Lattice QCD Code Generation

Denis Barthou, Gilbert Grosdidier+3 more|arXiv (Cornell University)|Aug 16, 2012

Parallel Computing and Optimization Techniques参考文献 5被引用 5

一句话总结

QIRAL 是一种领域特定语言，可对格点量子色动力学（LQCD）算法和矩阵结构进行高层级描述，通过利用 LQCD 模拟中固有的规则稀疏矩阵结构，自动生成优化的并行代码。它将算法设计与底层并行化解耦，支持在多核及混合架构上自动生成功能代码，从而实现对新型预条件化方法和迭代求解器的快速探索。

ABSTRACT

Quantum chromodynamics (QCD) is the theory of subnuclear physics, aiming at mod- eling the strong nuclear force, which is responsible for the interactions of nuclear particles. Lattice QCD (LQCD) is the corresponding discrete formulation, widely used for simula- tions. The computational demand for the LQCD is tremendous. It has played a role in the history of supercomputers, and has also helped defining their future. Designing efficient LQCD codes that scale well on large (probably hybrid) supercomputers requires to express many levels of parallelism, and then to explore different algorithmic solutions. While al- gorithmic exploration is the key for efficient parallel codes, the process is hampered by the necessary coding effort. We present in this paper a domain-specific language, QIRAL, for a high level expression of parallel algorithms in LQCD. Parallelism is expressed through the mathematical struc- ture of the sparse matrices defining the problem. We show that from these expressions and from algorithmic and preconditioning formulations, a parallel code can be automatically generated. This separates algorithms and mathematical formulations for LQCD (that be- long to the field of physics) from the effective orchestration of parallelism, mainly related to compilation and optimization for parallel architectures.

研究动机与目标

• 解决在现代混合超级计算机上高效、可扩展地编写格点量子色动力学（LQCD）模拟代码日益增长的复杂性问题。
• 减少为适应不断演进的硬件架构而手动调优底层并行化和数据布局所需的工作量。
• 在无需重写底层代码的前提下，系统性地探索 LQCD 中的新算法和预条件化技术。
• 将 LQCD 问题的数学公式与并行化和优化的编排分离开来。
• 支持自动生成功能复杂、结构化稀疏矩阵的代码，包括经预条件化修改后的矩阵，这些矩阵在 LQCD 中常见。

提出的方法

• QIRAL 使用高级语言表达 LQCD 算法和矩阵定义，通过在密集矩阵上使用代数运算符。
• 通过直和与张量积表示稀疏矩阵，捕捉格点 QCD 问题的四维模板结构。
• 语言语义基于 Maude 中的重写规则，支持对数学表达式的正式操作。
• 重写策略将索引求和与求积转换为并行循环，并自动插入 OpenMP 指令。
• 编译器应用标准优化技术，如循环融合、依赖性分析和标量提升。
• 高层级矩阵运算（如矩阵-向量乘积）通过重写规则映射到库调用（如 BLAS 或自定义 LION 库）。

实验结果

研究问题

• RQ1 高级领域特定语言是否能有效表达格点 QCD 问题的数学结构，并实现优化并行代码的自动生成？
• RQ2 如何利用 LQCD 矩阵的规则稀疏结构，在无需低层编码的情况下表达并行性？
• RQ3 通过将算法设计与底层实现细节解耦，能在多大程度上加速 LQCD 中的算法探索？
• RQ4 相同的高层级规范是否能跨不同架构（包括多核和混合系统）生成正确且高效的代码？
• RQ5 当预条件化导致复杂矩阵结构形成并破坏简单模板模式时，该方法的可扩展性如何？

主要发现

• QIRAL 能够从 LQCD 问题的高层级数学规范成功生成正确且优化的 C 代码，并带有 OpenMP 并行性。
• 该语言支持超越简单模板的复杂矩阵结构，包括经预条件化修改的矩阵，而先前工具（如 Pochoir）则不具备此能力。
• 自动代码生成支持在 4⁴ 网格上快速比较不同迭代求解器（如 CGNR、BiCGSTAB、MCR1）的性能，观察到收敛速度的差异。
• 该方法将算法开发与底层性能调优解耦，显著降低了探索新型预条件化和求解策略的成本。
• 编译器使用 Maude 中的重写机制，支持对代码变换的正式推理，并在优化过程中保持语义正确性。
• 生成的代码可通过额外的数据布局映射和库映射策略，进一步扩展以支持异构架构（如 GPU）。