[논문 리뷰] A GEMM-based direct solver for finite-difference Poisson problems in non-uniform grids

Pedro Filipe da Silva Costa, Duarte Palancha외 3명|arXiv (Cornell University)|2026. 03. 10.

Parallel Computing and Optimization Techniques인용 수 0

한 줄 요약

이 논문은 2개 방향에서 연산자를 대각화하고 세 번째 방향을 직접 해결하는 3D 비균일 격자용 직접 Poisson 해법을 제시하며, 전/후 변환은 CPU/GPU 효율성을 위해 GEMM으로 구현된다.

ABSTRACT

We present a direct Poisson solver for massively parallel simulations on three-dimensional Cartesian grids with non-uniform spacing. The method uses a tensor-based formulation in which the operator is diagonalized numerically along two directions through one-dimensional eigendecompositions, while the third direction is solved directly. The resulting dense transforms are evaluated efficiently as GEMMs (General Matrix--Matrix Multiplications), allowing many independent one-dimensional operations to be combined into matrix-matrix products that map well to modern CPU and GPU hardware. For uniform grids, the method reduces to the classical eigenfunction-expansion approach, and it naturally supports hybrid combinations of FFT-based and GEMM-based transforms depending on grid uniformity. After coupling the solver to an incompressible Navier-Stokes code, we assess its accuracy and performance against geometric multigrid and block cyclic reduction with FFT diagonalization. The results show that the proposed method is robust and consistently achieves the best time-to-solution. In strong scaling, the more compute-intensive GEMM-based variants attain higher parallel efficiency by better amortizing communication costs, while weak scaling highlights the expected trade-off between FFT-based and dense-transform formulations. Overall, the method enables efficient high-resolution stretched-mesh simulations on modern heterogeneous systems.

연구 동기 및 목표

• DNS의 비압축성 흐름에서 발산해제(incompressibility) 를 강제하기 위한 빠른 Poisson 해법의 필요성에 대한 동기 부여.
• eigenfunction 기반의 직접 해법을 비균일 격자까지 확장하면서도 GPU 친화적 커널을 유지.
• 기존 CaNS 인프라에 통합되고 균일 방향은 FFT, 비균일 방향은 GEMM으로 구성된 하이브리드 합성을 지원하는 GEMM 기반 구현 개발.
• 최첨단 방법(기하학적 다중격자 및 FFT를 이용한 블록 순환 축소) 대비 성능 평가.
• 고해상도 늘린(mesh)에서 CPU와 GPU의 강/약 규모 확장성(strong and weak)을 시연.

제안 방법

• 비균일 격자에서 1D 포아송 연산자를 삼대삼대 삼항대칭 시스템으로 형식화하고 대각 스케일링 D를 통해 대칭 유사체 T̃를 얻으며 T̃ = D^{1/2} T D^{-1/2}로 변환한다.
• 대칭 유사체 T̃ = Q̃ Λ Q̃^T의 고유분해를 수행하고 Q를 Q = D^{-1/2} Q̃에서 구하고 Q^{-1}를 Q̃^T D^{1/2}로 얻는다.
• 3D로 확장하기 위해 Kronecker 합: T = T_x ⊗ I ⊗ I + I ⊗ T_y ⊗ I + I ⊗ I ⊗ T_z; x 및 y(T_x, T_y) 방향으로 대각화를 수행하고 z 방향에서 독립적인 1D 삼항대각 문제를 푼다.
• 전방 고유 기저 변환과 역전변환은 x·y 방향에서 GEMM(또는 균일 방향의 FFT)으로 수행되고, z 방향에서는 직접 삼항대각 풀이(TDMA)가 사용된다.
• 이 방법은 FFT 기반 대각화와 동일한 도메인 분해 및 통신 패턴을 보존하여 균일 격자에는 FFT, 비균일 격자에는 GEMM으로 구성된 하이브리드 합성을 가능하게 한다.
• 포아송 해법은 분수 스텝 RK3 스킴과 함께 시스템 내에서 Helmholtz/Poisson 프레임워크를 통해 선택적 내향확산을 동일한 Poisson 기반 구조로 해결하는 Navier–Stokes 해법과 통합된다.

실험 결과

연구 질문

• RQ1 GEMM 기반의 두 방향 대각화가 FFT 기반 및 다중격자 접근법과 비교하여 비균일 격자에서 3D 포아송 문제의 솔루션 시간에 대해 경쟁적이거나 우월한가?
• RQ2 늘어난 방향으로의 성장에서 CPU와 GPU에서의 강/약 확장성 면에서 하이브리드 FFT/GEMM 접근법의 성능은 어떤가?
• RQ3 문제 크기가 커질 때 GEMM 기반 변환의 유연성(비균일 격자)과 점근적 비용 사이의 트레이드오프는 무엇인가?
• RQ4 비압축성 Navier–Stokes 워크플로우와의 solver 통합 및 현대 하드웨어에서 비균일 격자 기능의 오버헤드는 어느 정도인가?

주요 결과

• GEMM 기반 접근법은 테스트된 대안 중 단일 CPU 코어에서 가장 짧은 솔루션 소요 시간(core time)과 특히 급도( strongly graded) 격자에서 가장 좋게 나타난다.
• CPU와 GPU에서의 강한 확장성은 GEMM 기반 변형이 FFT 위주 변형보다 전치 및 통신 오버헤드를 더 잘 상쇄한다는 것을 보여준다.
• 약한 확장성(weak scaling)에서는 트레이드오프가 나타나: 증가하는 방향에서 FFT를 사용하면 시간 증가가 덜 나타나고, GEMM 기반 변환은 해상도가 증가함에 따라 비용이 더 증가한다.
• GPU에서 전체 비균일 격자 기능은 중간 수준의 오버헤드만 발생하고 다수의 GPU에서도 강한 확장을 유지한다.
• 전반적으로 이 방법은 균일 격자 실행 대비 스트레치된(mesh)에서 난류 흐름 시뮬레이션의 실시간 벽시계 시간 절감을 가능하게 하며 이종 하드웨어에서도 견고한 성능을 보인다.