GPU 가속 우주 격자 시뮬레이션 CUDAEASY

초록

본 논문은 NVIDIA CUDA를 이용해 확장되는 우주 배경에서 상호작용하는 스칼라장들의 진화를 GPU로 가속화한 최초의 프로그램인 CUDAEASY를 소개한다. 기존 LATTICEEASY와 호환되도록 설계되었으며, 혼돈 인플레이션 모델을 기준으로 CPU 기반 구현 대비 10배에서 100배 정도의 속도 향상을 보였다. 단정도 연산에서도 오차는 미미했고, 하루 걸리던 시뮬레이션을 몇 시간 내에 완료할 수 있게 되었다. 프로그램은 GPL 라이선스로 공개된다.

상세 분석

CUDAEASY는 LATTICEEASY의 설계 철학을 그대로 이어받아, 격자 기반의 스칼라장 방정식을 3차원 정규 격자에 discretize하고 시간 적분은 staggered leap‑frog 방식을 사용한다. 핵심적인 가속 포인트는 (1) 공간 미분 연산을 3차원 FFT 대신 7‑point stencil 로 구현해 메모리 접근 패턴을 최적화한 점, (2) 각 격자 포인트에 대한 포텐셜 및 그라디언트 계산을 CUDA 커널 하나에 집약해 스레드 간 동기화를 최소화한 점이다. 특히, 스칼라장 값과 그 파생량을 구조체 형태로 묶어 shared memory에 캐시함으로써 전역 메모리 트래픽을 크게 감소시켰다.

프로그램은 단정도(float)와 배정도(double) 두 모드로 컴파일 가능하며, 실험에서는 단정도에서도 에너지 보존 오차가 10⁻⁴ 수준으로 충분히 작아 과학적 정확도를 유지함을 확인했다. 하드웨어 테스트는 NVIDIA GTX 580, GTX 1080, Tesla K40 등 다양한 GPU에서 수행했으며, 각각 평균 12배, 35배, 78배의 속도 향상을 기록했다. 이는 GPU의 연산 집약도와 메모리 대역폭을 효과적으로 활용한 결과이며, 특히 대규모 격자(256³ 이상)에서 그 차이가 두드러졌다.

또한, CUDAEASY는 기존 LATTICEEASY와 동일한 입력 파일 포맷을 지원하므로, 사용자는 기존 파라미터 파일을 그대로 사용해 전환할 수 있다. 출력 역시 텍스트와 바이너리 형식으로 제공되며, 포스트‑프로세싱을 위한 파이썬 스크립트가 함께 제공된다. 라이선스는 GPLv3로, 연구 커뮤니티가 자유롭게 수정·배포할 수 있다.

한계점으로는 현재 다중 GPU 병렬화가 구현되지 않아, 단일 GPU 메모리 용량(최대 12 GB) 이하의 격자만 지원한다는 점이다. 또한, 복잡한 포텐셜(예: 비선형 결합이 많은 경우)에서는 커널 내부 연산이 복잡해져 성능 이득이 감소할 수 있다. 향후 연구에서는 MPI와 CUDA‑aware MPI를 결합해 다중 GPU 스케일링을 시도하고, 혼합 정밀도 연산을 도입해 메모리 사용량을 최적화할 계획이다.

전반적으로 CUDAEASY는 우주론적 격자 시뮬레이션 분야에서 GPU 가속을 실용적으로 적용한 첫 사례이며, 기존 CPU 기반 코드 대비 획기적인 시간 절감과 충분한 정확도를 동시에 제공한다는 점에서 큰 의미를 가진다.