CP2K on the road to exascale

기술 요약: 엑사스케일 시대를 향한 CP2K

문제 정의
응축물 물리학에서 양자 화학에 이르기까지 원자 단위 시뮬레이션을 위한 다재다有한 도구인 CP2K 프로그램 패키지는 엑사스케일 시대로의 전환에 따른 도전에 직면해 있다. CP2K는 역사적으로 병렬 확장성 측면에서 탁월한 성능을 보여왔으나, 엑사스케일 컴퓨팅으로의 전환을 위해서는 세 가지 특정 추세를 해결해야 한다: 대규모 병렬성(노드 및 스레드 수의 증가)의 필요성, GPU의 광범위한 채택, 그리고 혼합 정밀도 하드웨어 가속기의 등장이다. 핵심적인 계산 병목 현상은 대규모 오비탈 자유(orbital-free) 및 Kohn–Sham 밀도 범함수 이론(DFT) 계산, 그리고 파동 함수 기반 상관 방법(Hartree-Fock, MP2, RPA, GW 등)의 효율적인 처리를 포함한다. 구체적으로, 수백만 개의 원자를 포함하는 시스템에 대한 4중 중심 2전자 반발 적분(ERI)의 계산과 고윳값 문제의 풀이는 시스템 크기에 따라 선형 스케일링을 유지하면서도 동기화 오버헤드를 최소적으로 하고 저정밀도 산술을 정확도 손실 없이 활용하는 알고리즘을 요구한다.

방법론
본 논문은 미래의 아키텍처에 최적화하기 위한 CP2K의 다각적인 개발 전략을 개설한다:

1. Post-Hartree-Fock 방법을 위한 알고리즘 최적화:

   • 적분 스크리닝(Integral Screening): ERI의 스케일링을 $O(N^4)$에서 $O(N^2)$로, 나아가 선형 스케일링으로 줄이기 위해, 저자들은 Schwarz 부등식 스크리닝과 밀도 행렬 스크리닝을 사용한다. 후자는 가장 큰 밀도 행렬 요소($P_{max}$)를 사용하여 적분을 필터링하며, 이상적으로는 2세대 Car-Parrinello 분자 동역학(AIMD)에서 이전 타임스텝으로부터 전파된 밀도 행력을 활용한다.
   • 근사 방식(Approximation Schemes): 구현은 보조 밀도 행렬 방법(ADMM)과 항등식 분해(Resolution of Identity, RI) 접근법에 의존한다. ADMM은 더 작은 보조 밀도 행렬을 사용하여 정확한 Hartree-Fock 교환 에너지를 근사하며, 국소적 DFT 수준에서의 차이를 보정한다. RI는 Coulomb 메트릭의 Cholesky 분해를 통해 계산된 보조 가우시안 기저 집합을 사용하여 4중 중심 적분을 2중 및 3중 중심 적분으로 인수분해한다.
   • 선형 스케일링 솔버(Linear-Scaling Solvers): 코드는 대규모 희소 행렬의 역행렬과 제곱근을 계산하기 위해 행렬 부호 함수(matrix sign function), $sign(A) = A(A^2)^{-1/2}$를 활용한다. 이 함수는 또한 Kohn–Sham 행렬을 밀도 행렬 $P$로 정제(purify)하는 데 사용되어, 선형 스케일링 전자 구조 계산을 가능하게 한다.

2. 엑사스케일을 위한 아키텍처 적응:

   • 병렬성 및 스레딩: MPI 기반을 바탕으로, CP2K는 멀티 스레딩을 위해 OpenMP를 통합한다. 리덕션(reduction) 및 원자적 접근 충돌로 인한 병목 현상을 방방지하기 위해, 저자들은 행렬 블록 행을 스레드에 고정 할당하여 동기화 필요성을 제거하는 분산 블록 희소 행렬(Distributed Block-sparse Matrices, DBM) 라이브러리를 개발했다.
   • GPU 가속: GPU 지원은 라이브러리 수준(예: DBM, COSMA, SpFFT, DBCSR)에서 구현된다. 저자들은 병목 현상이 레지스터 파일 제약에서 호스트-디바이스 간 통신으로 변화하고 있음을 언급하며, 이를 GPU-aware MPI를 통해 해결하고 있다.
   • 혼합 정밀도 및 서브매트릭스 방법(Submatrix Method): 저정밀도(예: FP32 누적을 사용하는 FP16)에서 높은 FLOPs를 제공하는 하드웨어 가속기를 활용하기 위해, 저자들은 서브매트릭스 방법을 도입한다. 이 접근법은 대규모 분산 희소 행렬에 대한 행렬 함수 평가를 많은 수의 조밀한(dense) 소형 서브매트릭스에 대한 병렬 평가로 변환한다. 이를 정제 기법(purification scheme)과 결합하면, 수치적 부정확성이 수정된 Langevin 유형 방정식에 의해 보상되는 근사 컴퓨팅 패러다임을 사용할 수 있다.

주요 기여 및 결과

• 고급 방법의 구현: 본 논문은 post-HF 방법(HF, MP2, RPA, hybrid/double-hybrid DFT, GW) 내에서 임의의 경계 조건에 대한 해석적 핵 그래디언트(analytic nuclear gradients) 구현을 상세히 설명하며, 이는 정확한 유한 온도 시뮬레이션의 전제 조건이다.
• 확장성 입증: double-$\zeta$ valence polarization 기저 집합을 사용한 선형 스케일링 접근법을 통해, 저자들은 Cray XT5에서 최대 100만 개의 원자를 포함하는 시스템을 시뮬레이션할 수 있음을 입증했다.
• 기록적인 성능: 서브매트릭스 방법과 정제 기법을 결합함으로써, 연구팀은 1억 개 이상의 원자를 포함하는 시스템에 대해 기록적인 규모의 선형 스케일링 전자 구조 계산을 달성했다. 이 노력은 324 PFLOPs의 지속 성능과 67% 이상의 효율을 기록했다.
• 소프트웨어 생태계: 현대적인 Fortran 2008을 활용한 CP2K의 모듈형 구조는 독립형 라이브러리(예: DBCSR)의 추출과 GPU 가속 라이브러리의 통합을 용이하게 하여, 다양한 하드웨어에 대한 코드의 적응성을 높였다.

의의
본 논문은 CP2K를 "임박한 엑사스케일 시대"의 핵심 도구로 자리매김한다. 그 의의는 점점 더 복잡하고 정확한 시뮬레이션을 처리할 수 있는 능력뿐만 아니라, 하드웨어 트렌드에 대한 전략적 적응에 있다. 서브매트릭스 방법을 근사 컴퓨팅과 성공적으로 통합함으로써, CP2K는 저정밀도 하드웨어 아키텍처를 효과적으로 활용하는 과제를 해결한다. 또한, 저자들은 딥 뉴럴 네트워크(DNN)를 활용한 "on-the-fly 학습" 체계와 정확한 구성 상호작용(configuration interaction) 및 축소 밀도 행렬 범함수 이론을 위한 하이브리드 양자-고전 알고리즘의 향후 통합 로드맵을 제시하며, 컴퓨팅 패러다임이 진화함에 따라 CP2K가 원자 단위 시뮬레이션 방법론의 최전선에 머물 것임을 시사한다.