본 연구는 AI를 활용한 시뮬레이션 분야에서 IPU의 활용 가능성을 평가합니다. 특히, OpenFOAM 시뮬레이션 데이터를 활용하여 머신러닝 모델을 학습시키는 CFD 응용 프로그램을 분석합니다. Poplar SDK에서 제공하는 TensorFlow를 활용하여 IPU-POD16 플랫폼에 프로그램을 적용하고 성능 확장성을 조사했습니다.
popdist 라이브러리를 사용하여 호스트 측에서 IPU로 학습 데이터를 공급하는 성능 병목 현상을 해결하여 최대 34%의 속도 향상을 달성했습니다. IPU 간 통신 오버헤드로 인해 두 개의 IPU를 사용하는 것이 하나의 IPU를 사용하는 것보다 처리량을 개선하지 못했습니다.
IPU 수를 2개에서 16개로 늘려 처리량을 560.8 샘플/초에서 2805.8 샘플/초로 향상시켰습니다.