원문 캡처 · arxiv.org
원문 캡처 · arxiv.org

다중 GPU 통신 API의 메모리 동작을 명확히 정의하는 모델이 제안됐다. 논문은 AMD가 오픈소스로 공개한 rocSHMEM에 이 모델을 구현하고, 네트워크 데이터의 캐시 우회를 지원하도록 SHMEM API를 조정해 세 가지 통신 작업에서 성능이 최대 1.18~2.48배 향상됐다고 보고했다.

논문에 따르면 rocSHMEM, MSCCL++, NCCL/RCCL GIN 같은 장치 측 통신 API는 메모리 시스템과 통신 연산이 상호작용하는 방식을 엄밀하게 규정한 메모리 모델이 부족하다. 이 때문에 API를 이용해 네트워크 통신을 최적화하는 작업은 오류가 나기 쉽고, 호환성과 상호운용성 문제를 겪을 수 있다.

연구진은 하위 메모리 시스템의 세부 사항을 API 사용자에게 감추고 개발을 단순화하는 API 수준 메모리 모델을 제안했다. 이 모델을 바탕으로 한 SHMEM API 변경은 네트워크 데이터의 캐시 우회를 지원하며, 논문은 다음과 같은 성능 향상을 보고했다.

  • 집단 통신 작업: 최대 2.48배
  • 연산과 통신을 융합한 작업: 최대 1.77배
  • 최신 전문가 병렬 통신 작업: 최대 1.18배

연구진은 모델의 형식적 완화가 잘 정의된 표준 인터페이스를 유지하면서도 저수준 대규모 언어 모델 성능 최적화를 가능하게 한다고 설명한다. 수치는 논문 초록에 제시된 각 작업의 최대 향상치다.

논문 초록 · PDF · DOI