이 연구에서는 2차원 이징 모델을 웨이퍼 스케일 엔진(WSE)에 구현하고 최적화하였다. 이를 위해 체커보드 알고리즘을 WSE 아키텍처에 맞게 조정하였으며, 16개의 스핀을 하나의 int16 워드로 압축하여 표현하고 효율적인 비트 연산을 활용하였다. 또한 인접한 처리 장치 간 통신을 최소화하는 8개의 배열 표현을 사용하여 거의 완벽한 약한 확장성을 달성하였다.
이러한 최적화를 통해 최대 754개의 시뮬레이션을 동시에 처리할 수 있으며, 최대 200백만 개의 스핀을 가진 이징 모델에 대해 초당 61.8조 번의 플립 시도를 달성할 수 있었다. 이는 NVIDIA V100 단일 장치 대비 최대 148배, NVIDIA H100 대비 최대 88배 향상된 성능이다. 이 결과는 WSE가 재료 모델링 분야를 포함한 과학 계산 분야에서 큰 잠재력을 가지고 있음을 보여준다.
A otro idioma
del contenido fuente
arxiv.org
Ideas clave extraídas de
by Dirk Van Ess... a las arxiv.org 04-29-2024
https://arxiv.org/pdf/2404.16990.pdfConsultas más profundas