[논문 리뷰] QIRAL: A High Level Language for Lattice QCD Code Generation

Denis Barthou, Gilbert Grosdidier외 3명|arXiv (Cornell University)|2012. 08. 16.

Parallel Computing and Optimization Techniques참고 문헌 5인용 수 5

한 줄 요약

QIRAL은 레이티스 양자색역학(QCD) 알고리즘과 행렬 구조를 고수준으로 기술할 수 있는 도메인 특화 언어이며, QCD 시뮬레이션에 내재된 규칙적이고 희박한 행렬 구조를 활용하여 최적화된 병렬 코드를 자동으로 생성한다. 이는 알고리즘 설계를 저수준 병렬화에서 분리함으로써, 다중 코어 및 하이브리드 아키텍처에서 새로운 조절법 및 반복적 해법의 빠른 탐색을 가능하게 한다.

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에서 발생하는 복잡한 구조적 희박 행렬(예: 조절에 의해 수정된 경우 포함)에 대한 자동 코드 생성을 지원하기 위해.

제안 방법

• QIRAL은 조밀 행렬에 대한 대수적 연산자를 사용하여 LQCD 알고리즘과 행렬 정의를 고수준 언어로 표현한다.
• 희박 행렬은 직접 합과 텐서 곱을 통해 표현되며, 레이티스 QCD 문제의 4차원 스텐실 구조를 포괄한다.
• 언어의 의미 체계는 Maude 기반의 재작성 규칙에 기반하여 수학적 표현의 형식적 다룰 수 있도록 한다.
• 재작성 전략은 인덱스가 붙은 합과 곱을 병렬 루프로 변환하며, OpenMP 지시어는 자동으로 삽입된다.
• 컴파일러는 루프 융합, 의존성 분석, 스칼라 프omotion과 같은 표준 최적화 기법을 적용한다.
• 고수준 행렬 연산(예: 행렬-벡터 곱)은 재작성 규칙을 통해 라이브러리 호출(예: BLAS 또는 고유의 LION 라이브러리)로 매핑된다.

실험 결과

연구 질문

• RQ1 고수준 도메인 특화 언어가 레이티스 QCD 문제의 수학적 구조를 효과적으로 표현하고 최적화된 병렬 코드를 자동 생성할 수 있는가?
• RQ2 LQCD 행렬의 규칙적이고 희박한 구조를 어떻게 활용하여 저수준 코딩 없이도 병렬성을 표현할 수 있는가?
• RQ3 알고리즘 설계를 저수준 구현 세부 사항에서 분리함으로써 LQCD에서의 알고리즘 탐색이 얼마나 빨라질 수 있는가?
• RQ4 동일한 고수준 사양이 다중 코어 및 하이브리드 시스템을 포함한 다양한 아키텍처에서 정확하고 효율적인 코드를 생성할 수 있는가?
• RQ5 조절에 의해 단순한 스텐실 패턴이 깨지는 복잡한 행렬 구조에 대해 이 접근법은 어떻게 확장 가능한가?

주요 결과

• QIRAL은 LQCD 문제의 고수준 수학적 사양으로부터 정확하고 최적화된 C 코드를 OpenMP 병렬화와 함께 성공적으로 생성한다.
• 이전 도구들(예: Pochoir)과 달리, 단순한 스텐실을 넘어서 조절에 의해 수정된 복잡한 행렬 구조도 지원한다.
• 자동 코드 생성 덕분에 4⁴ 격자에서 다양한 반복적 해법(CGNR, BiCGSTAB, MCR1 등) 간의 수렴 속도 차이를 신속하게 비교할 수 있었다.
• 이 방법은 알고리즘 개발과 저수준 성능 튜닝을 분리함으로써, 새로운 조절법 및 해법 전략을 탐색하는 데 드는 비용을 크게 감소시켰다.
• 컴파일러가 Maude에서의 재작성 기법을 사용함으로써 코드 변환에 대한 형식적 추론이 가능해졌고 최적화 과정에서 의미적 정확성이 유지되었다.
• 생성된 코드는 추가적인 데이터 레이아웃 및 라이브러리 매핑 전략을 통해 GPU를 포함한 이종 아키텍처 지원으로도 확장 가능하다.