NVIDIA DGX Spark에서 Ling-3.0-flash 추론을 빠르게 하는 MoE 연산 최적화 방법
3 Ways to Boost MoE Inference on NVIDIA DGX Spark
NVIDIA 기술 블로그는 Inclusion AI의 Ling-3.0-flash 모델을 NVIDIA DGX Spark에서 batch-size-1로 평가한 결과를 다룬다. 프로파일링 결과 디코딩은 메모리 바운드였고, 모델 가중치를 읽는 데 실행 시간이 주로 쓰였다. 이 중 혼합 전문가(MoE) 연산과 점수를 산출하는 출력 레이어 프로젝션이 두드러진 병목으로 확인됐다.
Ling-3.0-flash 추론의 병목이 메모리 바운드 디코딩과 MoE 연산, 출력 레이어 프로젝션에 있다는 점을 확인했으므로, DGX Spark에서 MoE 모델을 서빙하는 개발자는 이 두 지점을 우선 점검하면 된다.
원문 출처
NVIDIA Technical Blog