[논문 리뷰] Generating SU(Nc) pure gauge lattice QCD configurations on GPUs with CUDA and OpenMP

Nuno Cardoso, Pedro Bicudo|arXiv (Cornell University)|2011. 12. 20.

Quantum Chromodynamics and Particle Interactions인용 수 2

한 줄 요약

이 논문은 GPU에서 SU(Nc) 순수 게이지 라티스 QCD 구성 생성을 위한 최적화된 CUDA 및 OpenMP 구현을 제시하며, SU(2), SU(3), SU(4) 및 Nc ≥ 4인 일반적인 SU(Nc) 코드에 대해 높은 성능을 달성한다. 저자들은 CPU 기반 방법 대비 뚜렷한 성능 향상을 입증하였으며, 라티스 QCD 공동체에서 공개 이용 가능한 소스 코드를 제공한다.

ABSTRACT

The starting point of any lattice QCD computation is the generation of a Markov chain of gauge field configurations. Due to the large number of lattice links and due to the matrix multiplications, generating SU(Nc) lattice QCD configurations is a highly demanding computational task, requiring advanced computer parallel architectures such as clusters of several Central Processing Units (CPUs) or Graphics Processing Units (GPUs). In this paper we present and explore the performance of CUDA codes for NVIDIA GPUs to generate SU(Nc) lattice QCD pure gauge configurations. Our implementation in one GPU uses CUDA and in multiple GPUs uses OpenMP and CUDA. We present optimized CUDA codes SU(2), SU(3) and SU(4). We also show a generic SU(Nc) code for Nc$\,\geq 4$ and compare it with the optimized version of SU(4). Our codes are publicly available for free use by the lattice QCD community.

연구 동기 및 목표

• 매트릭스 곱셈과 큰 라티스 크기로 인해 계산적으로 비용이 많이 드는 SU(Nc) 순수 게이지 라티스 QCD 구성 생성을 가속화하기 위해.
• NVIDIA GPU에서 데이터 수준 병렬성을 활용하기 위해 CUDA를 사용한 단일 GPU 성능을 확보하고, 확장성을 위해 OpenMP를 활용한 다중 GPU 시스템으로 확장하기 위해.
• Nc ≥ 4인 경우에 대해 최적화된 일반적인 SU(Nc) 코드를 개발하여 SU(3)과 SU(2)를 초월한 효율적인 시뮬레이션을 가능하게 하기 위해.
• 라티스 QCD 공동체가 라티스 양자장 이론 시뮬레이션을 발전시키는 데 지원할 수 있도록 공개 이용 가능한 고성능 소스 코드를 제공하기 위해.

제안 방법

• NVIDIA GPU에서 데이터 병렬성을 활용하기 위해 CUDA 커널을 사용한 SU(Nc) 게이지 장 업데이트 구현.
• 하이브리드 CPU-GPU 아키텍처에서 다중 GPU 간 데이터 분배 및 동기화를 관리하기 위해 OpenMP 활용.
• GPU 스트리밍 다중처리기에서 지연을 최소화하고 할당률을 극대화하기 위해 메모리 접근 패턴과 커널 실행 설계.
• 장치 특화 CUDA 기능을 활용한 매트릭스 곱셈 및 SU(Nc) 군 투영 루틴 최적화.
• 특정 군 구조에서 독립된 일반적인 SU(Nc) 커널 개발 (Nc ≥ 4 지원).
• 단일 및 다중 GPU 시스템에서 SU(2), SU(3), SU(4) 구성에 대한 성능 벤치마킹.

실험 결과

연구 질문

• RQ1 CUDA를 통한 GPU 가속화가 어떻게 SU(Nc) 라티스 QCD 구성 생성의 성능을 향상시킬 수 있는가?
• RQ2 OpenMP와 CUDA를 조합한 다중 GPU 사용이 라티스 QCD 시뮬레이션에서 확장성과 성능 향상에 어떻게 기여하는가?
• RQ3 Nc ≥ 4인 일반적인 SU(Nc) 구현의 성능은 Nc = 4인 최적화된 SU(4) 코드와 비교해 어떻게 되는가?
• RQ4 GPU 기반 라티스 게이지 장 생성에서 주요 성능 저하 요인은 무엇이며, 이를 어떻게 완화할 수 있는가?

주요 결과

• CUDA 최적화 단일 GPU 구현은 SU(2), SU(3), SU(4) 게이지 군에 대해 전통적인 CPU 기반 방법 대비 뚜렷한 성능 향상을 달성한다.
• 하이브리드 OpenMP + CUDA 접근 방식은 다중 GPU 간 효율적인 확장성을 가능하게 하여 큰 라티스에 대한 구성 생성 시간을 크게 단축시킨다.
• Nc ≥ 4인 일반적인 SU(Nc) 코드는 SU(3)을 초월한 임의의 유니터리 게이지 군을 지원하면서도 높은 성능을 유지한다.
• 성능 벤치마킹 결과, 최적화된 SU(4) 커널은 Nc = 4인 일반적인 SU(Nc) 구현보다 뛰어난 성능을 보이며, 코드 특화의 가치를 입증한다.
• 저자들은 라티스 QCD 공동체에 공개 이용 가능한, 생산용으로 사용 가능한 소스 코드를 성공적으로 기여하여 고성능 시뮬레이션에 대한 광범위한 접근성을 제공한다.