跳至主要內容
標籤工程技術誌

#GPU Memory Management

2 篇文章

MoE 分片:混合專家模型的平行化策略

在 I-00 篇中,我們列出了 LLM 推論與傳統模型服務的五項差異。前四項——可變長度運算、兩階段資源特徵、不斷增長的記憶體需求,以及快取感知路由——每一項都已經在本系列的專文中討論過了。第五項則只用一句話帶過:「有些現代 LLM 使用混合專家模型(mixture-of-experts, MoE)架構,模型的不同部分會因不同的輸入而被啟用。把 MoE 模型分散到多個 GPU 上,需要跟密集模型不同的分片(sharding)策略。」

Huang Tzu Lin

分頁式 KV 快取:LLM 推論服務的 GPU 記憶體管理

在 I-00 篇中,我們走過了一次 API 呼叫從頭到尾通過推論流程的完整路徑,也認識了 KV 快取(key-value cache)——一種用來儲存注意力機制中 key-value 向量的資料結構,讓模型不必在每個解碼步驟重複計算這些向量。KV 快取會隨著每個生成的 token 不斷增長,而且在整個請求期間都必須留在 GPU 記憶體裡。到了 I-01 篇,我們又認識了連續批次處理(continuous batching):它在迭代層級進行排程,不必等批次中最慢的請求跑完,因此能讓更多請求同時保持運作。

Huang Tzu Lin