하드웨어에 구애받지 않는 플라즈마 PIC 코드 Entity

초록

Entity는 Kokkos 기반의 하드웨어‑agnostic 설계를 통해 CPU·GPU 전천후 성능 이식성을 제공하는 최신 입자‑인‑셀(PIC) 코드이다. 일반 좌표계(직교, 축대칭 구면, 준구면 등)에서 Vlasov‑Maxwell 방정식을 풀며, 입자 운동은 전역 직교 기준으로 수행해 수치적 안정성을 확보한다. 전하 보존은 √h 가중 전류(Conformal Current) 기법으로 구현하고, 3‑D 입자 푸셔와 전류 삽입이 2 ns/입자·스텝 수준의 효율을 보인다. 다양한 GPU(AMD MI250X, NVIDIA A100, Intel Max)에서 확장성·스케일링을 검증했으며, 표준 Cartesian 테스트와 축대칭 천체 자기장 시뮬레이션으로 정확성을 입증한다.

상세 분석

Entity는 현대 천체 플라즈마 시뮬레이션이 직면한 두 가지 근본적 한계—극단적인 규모 격차와 GPU‑중심 컴퓨팅 환경—를 동시에 해결하도록 설계되었다. 첫 번째 핵심은 Kokkos 라이브러리를 활용한 하드웨어‑agnostic 추상화이다. Kokkos의 실행 정책과 메모리 레이아웃을 통해 CPU, NVIDIA, AMD, Intel GPU에서 동일한 소스 코드를 거의 수정 없이 컴파일·실행할 수 있다. 이는 기존 PIC 코드가 CPU 전용으로 설계된 경우와 달리, 새로운 GPU 아키텍처가 등장해도 코드 재작성 없이 성능을 유지한다는 큰 장점을 제공한다.

두 번째 핵심은 일반 좌표계에서 Vlasov‑Maxwell 방정식을 푸는 좌표‑agnostic 프레임워크이다. 저자들은 메트릭 텐서가 대각인 경우(직교, 축대칭 구면 등)만을 대상으로 하지만, 좌표 변환을 위한 tetrad(사변) 행렬을 명시적으로 도입해 모든 물리량을 전역 직교 기준으로 변환한다. 입자 운동 방정식은 전역 직교 좌표계에서 Boris‑type 푸셔를 그대로 적용하고, 좌표 변환만 추가함으로써 기존 입자 푸셔의 수치적 안정성을 그대로 유지한다. 이는 곡률 항으로 인한 수치 발산 위험을 회피하고, 고차원 곡면에서도 시간 적분 안정성을 보장한다.

전하 보존은 √h 로 가중된 전류(Conformal Current) 개념을 도입해 해결한다. 일반 좌표계에서 전류 밀도 J_i는 √h 로 스케일링된 형태로 정의되며, 기존의 charge‑conserving current deposition 알고리즘(예: Esirkepov, Villasenor‑Buneman)을 그대로 적용할 수 있다. 이후 실제 물리 전류는 J_i/√h 로 복원해 Maxwell‑Ampère 방정식에 삽입한다. 이 접근법은 복잡한 좌표계에서도 전하 보존을 엄격히 만족시키며, 기존 코드와의 호환성을 크게 높인다.

성능 측면에서 저자들은 3‑D 입자 푸셔와 전류 삽입 단계가 입자당 2 ns/스텝 수준의 처리 속도를 보인다고 보고한다. 이는 최신 GPU에서 입자당 연산이 메모리 바운드가 아닌 연산 바운드임을 의미한다. 또한, 전체 PIC 루프에서 메모리 사용을 최소화하기 위해 단계별로 필요한 배열만 유지하고, 필터링 단계에서도 인라인 SIMD 최적화를 적용했다. 강력한 스케일링 테스트에서는 1 M 입자·10 M 격자 규모에서 4096 GPU까지 거의 선형 확장성을 보였으며, 플랫폼 간 성능 차이는 10 % 이내에 머물렀다.

검증 사례는 두 부류로 나뉜다. 첫 번째는 전통적인 Cartesian 테스트(Weibel 불안정성, 두 플라즈마 스트림, 전자/양성자 충돌 등)로, 기존 PIC 코드와 비교해 오차가 10⁻⁴ 이하임을 확인했다. 두 번째는 축대칭 구면 좌표에서의 상대론적 자기장 시뮬레이션으로, 중성자 별 및 블랙홀 주변의 강자기장 구조를 재현했다. 특히, 구면 좌표에서의 전류 삽입과 필터링이 정확히 전하 보존을 유지하면서도 물리적 스펙트럼을 보존함을 입증했다.

전반적으로 Entity는 (1) 하드웨어 독립적인 고성능 포팅, (2) 일반 좌표계에서의 수치 안정성 확보, (3) 모듈식 API를 통한 사용자 정의 가능성, (4) 검증된 물리 정확도라는 네 가지 축을 중심으로 현대 천체 플라즈마 연구에 필요한 모든 요소를 제공한다는 점에서 큰 의미를 가진다. 향후 일반 상대론적(중력) 좌표계, 고차원 입자 형태 함수, QED·핵반응 모듈 등 확장이 예정돼 있어, 다중 물리 현상을 포괄하는 통합 시뮬레이션 플랫폼으로 자리매김할 가능성이 높다.