ArborX 2.0 새로운 인터페이스와 분산 검색 기능

초록

**
ArborX 2.0은 Kokkos 기반의 성능 이식성 지오메트리 검색 라이브러리로, API를 전면 개편해 사용자 정의 객체와 콜백을 지원하고, 브루트포스·분산 트리 구조를 추가했으며, 레이 트레이싱·클러스터링 등 새로운 알고리즘을 제공한다.

**

상세 분석

**
ArborX 2.0은 기존 BVH 기반 API(v1)의 한계를 극복하기 위해 근본적인 설계 변화를 도입했다. 첫 번째 변화는 템플릿 파라미터의 확장이다. 메모리 공간뿐 아니라 저장값(Value), 인덱스 가능한 객체(IndexableGetter), 바운딩 볼륨(BoundingVolume)을 모두 템플릿 인자로 받아, 1‑10 차원의 다양한 기하학적 형상(k‑DOP, 삼각형, 레이, 테트라헤드론 등)을 직접 다룰 수 있게 되었다. 두 번째는 실행 공간(execution space)의 명시적 전달이다. 생성자와 query 함수 모두 Kokkos 실행 공간을 인자로 받아, 동일 디바이스에서 여러 검색을 병렬로 수행하거나, 사용자 커스텀 커널과 동시에 실행할 수 있다. 이는 GPU‑aware MPI와 결합했을 때 스트림 기반 동시 실행을 가능하게 하여, 기존의 전역 펜싱 문제를 해소한다.

세 번째 핵심은 콜백 메커니즘이다. 순수 콜백(query‑only)과 결과 저장 콜백(query‑with‑output) 두 형태를 제공해, 검색 결과를 메모리에 복사하지 않고 바로 사용자 정의 연산(예: 거리 누적, 평균 계산, 필드 보간 등)을 수행한다. 콜백은 조기 종료 신호를 반환할 수 있어, N개까지만 결과를 수집하거나 특정 조건에서 탐색을 중단하는 고급 제어가 가능하다.

네 번째는 분산 트리(DistributedTree) 의 도입이다. MPI 커뮤니케이터를 받아 로컬 BVH를 구축하고, 상위 레벨 트리로 로컬 바운딩 박스를 요약한다. 검색 시 각 프로세스는 필요한 쿼리만 다른 프로세스로 라우팅하고, 콜백은 데이터를 소유한 프로세스에서 직접 실행된다. GPU‑aware MPI 옵션을 활성화하면 데이터가 GPU 메모리 상에 머무른 채 통신이 이루어져, 호스트‑디바이스 복사의 오버헤드를 크게 감소시킨다.

다섯 번째는 새로운 검색 구조이다. 기존 BVH 외에 브루트포스 구조와 분산 트리를 제공해, 데이터 규모가 작거나 메모리 대역폭이 제한된 상황에서도 효율적인 검색이 가능하도록 설계되었다.

마지막으로 알고리즘 확장이 눈에 띈다. 레이 트레이싱, DBSCAN·FDBSCAN·FDBSCAN‑DenseBox 기반 밀도 기반 클러스터링, GPU 최적화 Euclidean Minimum Spanning Tree(EMST) 등 과학·공학 워크플로우에 바로 적용 가능한 기능을 추가했다. 특히 DBSCAN 구현은 희소·밀집 데이터에 맞춰 두 가지 변형을 제공함으로써, 대규모 천문학 시뮬레이션에서도 실시간 클러스터링이 가능하도록 설계되었다.

전반적으로 ArborX 2.0은 유연성, 확장성, 성능을 동시에 만족시키는 설계로, 기존 API와의 호환성을 포기하고라도 현대 HPC 환경(다중 GPU, MPI, 다양한 기하학)에서의 실용성을 크게 향상시켰다.

**