亚洲天堂无码专区精品影院精品一区,青青草在线视频成人

在上篇文章中，已經(jīng)說(shuō)明了有狀態(tài)（stateful）模型的“控制輸入”與“隱式狀態(tài)管理”的使用方式，本文內(nèi)容接著就繼續(xù)說(shuō)明“調(diào)度策略”的使用。（續(xù)前一篇文章的編號(hào)） (3) 調(diào)度策略（Scheduling Strategies）在決定如何對(duì)分發(fā)到同一模型實(shí)例的序列進(jìn)行批處理時(shí)，序列批量處理器（sequence batcher）可以采用以下兩種調(diào)度策略的其中一種：

直接（direct）策略

當(dāng)模型維護(hù)每個(gè)批量處理槽的狀態(tài)，并期望給定序列的所有推理請(qǐng)求都分發(fā)到同一槽，以便正確更新?tīng)顟B(tài)時(shí)，需要使用這個(gè)策略。此時(shí)，序列批量處理程序不僅能確保序列中的所有推理請(qǐng)求，都會(huì)分發(fā)到同一模型實(shí)例，并且確保每個(gè)序列都被分發(fā)至模型實(shí)例中的專用批量處理槽（batch slot）。下面示例的模型配置，是一個(gè) TensorRT 有狀態(tài)模型，使用直接調(diào)度策略的序量批處理程序的內(nèi)容：

name: "direct_stateful_model"platform: "tensorrt_plan"max_batch_size: 2sequence_batching{ max_sequence_idle_microseconds: 5000000direct { } control_input [{name: "START" control [{ kind: CONTROL_SEQUENCE_START fp32_false_true: [ 0, 1 ]}]},{name: "READY" control [{ kind: CONTROL_SEQUENCE_READY fp32_false_true: [ 0, 1 ]}]}]}#續(xù)接右欄

#上接左欄input [{name: "INPUT" data_type: TYPE_FP32dims: [ 100, 100 ]}]output [{name: "OUTPUT" data_type: TYPE_FP32dims: [ 10 ]}]instance_group [{ count: 2}]

現(xiàn)在簡(jiǎn)單說(shuō)明以下配置的內(nèi)容：

sequence_batching 部分指示模型會(huì)使用序列調(diào)度器的 Direct 調(diào)度策略；
示例中模型只需要序列批處理程序的啟動(dòng)和就緒控制輸入，因此只列出這些控制；
instance_group 表示應(yīng)該實(shí)例化模型的兩個(gè)實(shí)例；
max_batch_size 表示這些實(shí)例中的每一個(gè)都應(yīng)該執(zhí)行批量大小為 2 的推理計(jì)算。

下圖顯示了此配置指定的序列批處理程序和推理資源的表示：

每個(gè)模型實(shí)例都在維護(hù)每個(gè)批處理槽的狀態(tài)，并期望將給定序列的所有推理請(qǐng)求分發(fā)到同一槽，以便正確更新?tīng)顟B(tài)。對(duì)于本例，這意味著 Triton 可以同時(shí) 4 個(gè)序列進(jìn)行推理。使用直接調(diào)度策略，序列批處理程序會(huì)執(zhí)行以下動(dòng)作：

所識(shí)別的推理請(qǐng)求種類	執(zhí)行動(dòng)作
需要啟動(dòng)新序列	1. 有可用處理槽時(shí)：就為該序列分配批處理槽2. 無(wú)可用處理槽時(shí)：就將推理請(qǐng)求放在積壓工作中
是已分配處理槽序列的一部分	將該請(qǐng)求分發(fā)到該配置好的批量處理槽
是積壓工作中序列的一部分	將請(qǐng)求放入積壓工作中
是最后一個(gè)推理請(qǐng)求	1. 有積壓工作時(shí)：將處理槽分配給積壓工作的序列2. 有積壓工作：釋放該序列處理槽給其他序列使用

下圖顯示使用直接調(diào)度策略，將多個(gè)序列調(diào)度到模型實(shí)例上的執(zhí)行：

圖左顯示了到達(dá) Triton 的 5 個(gè)請(qǐng)求序列，每個(gè)序列可以由任意數(shù)量的推理請(qǐng)求組成。圖右側(cè)顯示了推理請(qǐng)求序列是如何隨時(shí)間安排到模型實(shí)例上的，

在實(shí)例 0 與實(shí)例 1 中各有兩個(gè)槽 0 與槽 1；
根據(jù)接收的順序，為序列 0 至序列 3 各分配一個(gè)批量處理槽，而序列 4 與序列 5 先處于排隊(duì)等候狀態(tài)；
當(dāng)序列 3 的請(qǐng)求全部完成之后，將處理槽釋放出來(lái)給序列 4 使用；
當(dāng)序列 1 的請(qǐng)求全部完成之后，將處理槽釋放出來(lái)給序列 5 使用；

以上是直接策略對(duì)最基本工作原理，很容易理解。接下來(lái)要進(jìn)一步使用控制輸入張量與模型通信的功能，下圖是一個(gè)分配給模型實(shí)例中兩個(gè)批處理槽的兩個(gè)序列，每個(gè)序列的推理請(qǐng)求隨時(shí)間而到達(dá)，START 和 READY 顯示用于模型每次執(zhí)行的輸入張量值：

?隨著時(shí)間的推移（從右向左），會(huì)發(fā)生以下情況：

序列中第一個(gè)請(qǐng)求（Req 0）到達(dá)槽 0 時(shí)，因?yàn)槟Ｐ蛯?shí)例尚未執(zhí)行推理，則序列調(diào)度器會(huì)立即安排模型實(shí)例執(zhí)行，因?yàn)橥评碚?qǐng)求可用；
由于這是序列中的第一個(gè)請(qǐng)求，因此 START 張量中的對(duì)應(yīng)元素設(shè)置為 1，但槽 1 中沒(méi)有可用的請(qǐng)求，因此 READY 張量?jī)H顯示槽 0 為就緒。
推理完成后，序列調(diào)度器會(huì)發(fā)現(xiàn)任何批處理槽中都沒(méi)有可用的請(qǐng)求，因此模型實(shí)例處于空閑狀態(tài)。
接下來(lái)，兩個(gè)推理請(qǐng)求（上面的 Req 1 與下面的 Req 0）差不多的時(shí)間到達(dá)，序列調(diào)度器看到兩個(gè)處理槽都是可用，就立即執(zhí)行批量大小為 2 的推理模型實(shí)例，使用 READY 顯示兩個(gè)槽都有可用的推理請(qǐng)求，但只有槽 1 是新序列的開(kāi)始（START）。
對(duì)于其他推理請(qǐng)求，處理以類似的方式繼續(xù)。

以上就是配合控制輸入張量的工作原理。

最舊的（oldest）策略

這種調(diào)度策略能讓序列批處理器，確保序列中的所有推理請(qǐng)求都被分發(fā)到同一模型實(shí)例中，然后使用“動(dòng)態(tài)批處理器”將來(lái)自不同序列的多個(gè)推理批量處理到一起。使用此策略，模型通常必須使用 CONTROL_SEQUENCE_CORRID 控件，才能讓批量處理清楚每個(gè)推理請(qǐng)求是屬于哪個(gè)序列。通常不需要 CONTROL_SEQUENCE_READY 控件，因?yàn)榕幚碇兴械耐评矶紝㈦S時(shí)準(zhǔn)備好進(jìn)行推理。下面是一個(gè)“最舊調(diào)度策略”的配置示例，以前面一個(gè)“直接調(diào)度策略”進(jìn)行修改，差異之處只有下面所列出的部分，請(qǐng)自行調(diào)整：

直接（direct）策略

最舊的（oldest）策略

direct {}

oldest

{

max_candidate_sequences: 4

}

在本示例中，模型需要序列批量處理的開(kāi)始、結(jié)束和相關(guān) ID 控制輸入。下圖顯示了此配置指定的序列批處理程序和推理資源的表示。

使用最舊的調(diào)度策略，序列批處理程序會(huì)執(zhí)行以下工作：

所識(shí)別的推理請(qǐng)求種類	執(zhí)行動(dòng)作
需要啟動(dòng)新序列	嘗試查找具有候選序列空間的模型實(shí)例，如果沒(méi)有實(shí)例可以容納新的候選序列，就將請(qǐng)求放在一個(gè)積壓工作中
已經(jīng)是候選序列的一部分	將該請(qǐng)求分發(fā)到該模型實(shí)例
是積壓工作中序列的一部分	將請(qǐng)求放入積壓工作中
是最后一個(gè)推理請(qǐng)求	模型實(shí)例立即從積壓工作中刪除一個(gè)序列，并將其作為模型實(shí)例中的候選序列，或者記錄如果沒(méi)有積壓工作，模型實(shí)例可以處理未來(lái)的序列。

下圖顯示將多個(gè)序列調(diào)度到上述示例配置指定的模型實(shí)例上，左圖顯示 Triton 接收了四個(gè)請(qǐng)求序列，每個(gè)序列由多個(gè)推理請(qǐng)求組成：

這里假設(shè)每個(gè)請(qǐng)求的長(zhǎng)度是相同的，那么左邊候選序列中送進(jìn)右邊批量處理槽的順序，就是上圖中間的排列順序。最舊的策略從最舊的請(qǐng)求中形成一個(gè)動(dòng)態(tài)批處理，但在一個(gè)批處理中從不包含來(lái)自給定序列的多個(gè)請(qǐng)求，例如上面序列 D 中的最后兩個(gè)推理不是一起批處理的。以上是關(guān)于有狀態(tài)模型的“調(diào)度策略”主要內(nèi)容，剩下的“集成模型”部分，會(huì)在下篇文章中提供完整的說(shuō)明。

原文標(biāo)題：NVIDIA Triton 系列文章（11）：模型類別與調(diào)度器-2

文章出處：【微信公眾號(hào)：NVIDIA英偉達(dá)企業(yè)解決方案】歡迎添加關(guān)注！文章轉(zhuǎn)載請(qǐng)注明出處。

聲明：本文內(nèi)容及配圖由入駐作者撰寫或者入駐合作網(wǎng)站授權(quán)轉(zhuǎn)載。文章觀點(diǎn)僅代表作者本人，不代表電子發(fā)燒友網(wǎng)立場(chǎng)。文章及其配圖僅供工程師學(xué)習(xí)之用，如有內(nèi)容侵權(quán)或者其他違規(guī)問(wèn)題，請(qǐng)聯(lián)系本站處理。舉報(bào)投訴

英偉達(dá)

英偉達(dá)

+關(guān)注

關(guān)注
23

文章
4091

瀏覽量
99308

原文標(biāo)題：NVIDIA Triton 系列文章（11）：模型類別與調(diào)度器-2

文章出處：【微信號(hào)：NVIDIA-Enterprise，微信公眾號(hào)：NVIDIA英偉達(dá)企業(yè)解決方案】歡迎添加關(guān)注！文章轉(zhuǎn)載請(qǐng)注明出處。

91欧美超碰AV自拍|国产成年人性爱视频免费看|亚洲日韩欧美一厂二区入|人人看人人爽人人操aV|丝袜美腿视频一区二区在线看|人人操人人爽人人爱|婷婷五月天超碰|97色色欧美亚州A√|另类A√无码精品一级av|欧美特级日韩特级

搜索歷史

NVIDIA Triton 系列文章（11）：模型類別與調(diào)度器-2

評(píng)論