멀티 파티션 NUMA GPU에서의 LLM 커널 액세스 및 메모리 상호작용 특성 분석
요약
멀티 파티션 NUMA GPU 환경에서 LLM 커널의 메모리 액세스 및 데이터 상호작용 특성을 분석한 연구입니다. 메모리 트레이스 분석과 시뮬레이션을 통해 워크그룹 간 데이터 공유 패턴을 분류하고, 지연 시간 최적화를 위한 다양한 스케줄링 전략을 제안합니다.
핵심 포인트
- 멀티 파티션 GPU의 NUMA 특성이 LLM 커널 지연 시간에 미치는 영향 분석
- LLM 커널 피연산자를 전역, 부분, 개인 공유 패턴으로 분류
- 워크그룹별 고정 및 서브그룹 인지 공동 스케줄링 등 최적화 전략 제시
- 배치 인지 커널 프로그래밍 및 아키텍처 지원의 필요성 강조
대규모 언어 모델 (LLM) 워크로드는 컴퓨팅 및 메모리 용량을 확장하기 위한 경로로서 멀티 파티션 (multi-partition) GPU 사용을 촉진하지만, 이들의 비균등 메모리 액세스 (non-uniform memory access, NUMA) 특성과 파티션 간 통신은 경합을 증폭시키고 지역성 (locality)을 저하시켜 최적화되지 않은 커널 지연 시간 (kernel latency)을 초래할 수 있습니다. 이를 해결하기 위해, 우리는 최첨단 서빙 엔진의 가중치 투영 (weight projection), 전문가 혼합 (mixture-of-experts, MoE), 그리고 어텐션 (attention) 변형을 아우르는 성능 핵심 LLM 커널 구현을 분석하여 멀티 파티션 GPU에서의 데이터 액세스 패턴 특성을 제시합니다. 먼저, 워크그룹 (workgroup) 수준의 데이터 액세스 및 공유 동작을 도출하기 위한 메모리 트레이스 분석 방법론을 도입하고, 사이클 수준 시뮬레이터 (cycle-level simulator)를 사용하여 지연 시간에 미치는 지역성의 영향을 평가합니다. 이러한 도구들을 사용하여, 우리는 LLM 커널 피연산자 (operands)를 세 가지 워크그룹 간 공유 패턴 (전역 (global), 부분 (partial), 또는 개인 (private))으로 분류하며, 단순한 워크그룹별 고정 (per-workgroup pinning)부터 서브그룹 인지 공동 스케줄링 (subgroup-aware co-scheduling)에 이르기까지 카테고리에 따라 필요한 최적화 전략이 다름을 보여줍니다. 우리의 연구 결과는 멀티 파티션 GPU에서 배치 인지 (placement-aware) 커널 프로그래밍과 작업 및 데이터 지역성을 위한 더 스마트한 아키텍처 지원의 필요성을 강조합니다.
AI 자동 생성 콘텐츠
본 콘텐츠는 arXiv cs.AR의 원문을 AI가 자동으로 요약·번역·분석한 것입니다. 원 저작권은 원저작자에게 있으며, 정확한 내용은 반드시 원문을 확인해 주세요.
원문 바로가기