技術 08 / 服務部署書籍來源
PagedAttention 與區塊配置
問題負責人執行環境工程師
token 的邏輯順序,與實體區塊擺在哪裡無關。示意圖,非基準測試結果。商業問題與這份工作
服務的總記憶體其實夠用,卻浪費在過大或破碎的請求配置上,導致能同時服務的客戶數變少。負責人要管好 KV 區塊與排程,讓長度不一的請求更有效率地吃下這些容量。
系統裡改變了什麼
把 KV 狀態切成固定大小的實體區塊。再用一張邏輯區塊對照表,讓注意力即使在區塊於記憶體中不連續的情況下,仍能正確存取整個序列。
必須承擔的取捨
頁表與核心會增加實作成本。分頁屬於記憶體管理;它本身既不會近似注意力,也不會縮減邏輯上的上下文長度。
決策所需的證據
配置浪費比例、同時在線請求數、記憶體壓力、失敗次數,以及尾端延遲。
技術 19 / 服務部署書籍+一手補充
上下文平行
問題負責人分散式推論工程師
每張裝置負責一段上下文,並且必須把相關的遠端片段一併納入計算。示意圖,非基準測試結果。商業問題與這份工作
影片或長上下文的工作負載,所產生的注意力運算與狀態已經超出單張裝置能處理的量。負責人把與序列相關的工作切分到多張裝置上。公司要的是在期限內交出完整的影片或答案,同時仍保有整段輸入的資訊。
系統裡改變了什麼
對注意力工作的序列(上下文)維度進行切分。裝置之間會交換該注意力樣式所需的遠端狀態,而且通常會讓傳輸與運算重疊進行。
必須承擔的取捨
把序列切開,不代表各段之間就此彼此獨立。通訊成本、被複製的權重,以及模型本身的注意力實作,共同決定了實際效益。
決策所需的證據
記憶體峰值、注意力運算時間、裝置間傳輸量、總完成時間,以及品質。