加利福尼亞州圣克拉拉——Nvidia通過(guò)一個(gè)名為TensorRT LLM的新開源軟件庫(kù)，將其H100、A100和L4 GPU的大型語(yǔ)言模型(LLM)推理性能提高了一倍。

正如對(duì)相同硬件一輪又一輪改進(jìn)的基準(zhǔn)測(cè)試結(jié)果所證明的那樣，在從專用人工智能芯片中擠出盡可能好的性能時(shí)，軟件往往與硬件一樣重要。

“我們所做的很大一部分是硬件和軟件的結(jié)合，如今英偉達(dá)的軟件工程師比硬件工程師多，”英偉達(dá)超大規(guī)模和HPC計(jì)算業(yè)務(wù)副總裁兼總經(jīng)理Ian Buck告訴《EE時(shí)報(bào)》?！斑@是回到最初的CUDA的決定的一部分，也是交付不僅是一個(gè)帶有指令集的芯片，而且是一個(gè)完整的堆棧以滿足開發(fā)者需求的動(dòng)機(jī)。

他說(shuō)：“這提供了一個(gè)在各個(gè)層面進(jìn)行創(chuàng)新的機(jī)會(huì)：改變硬件架構(gòu)、改變指令集、改變編譯器、改變驅(qū)動(dòng)程序、改變工具、庫(kù)等等，這樣我們就可以推動(dòng)整個(gè)平臺(tái)向前發(fā)展?！??！霸谶^(guò)去20年的加速計(jì)算中，這種情況已經(jīng)多次出現(xiàn)，人工智能推理也是如此?！?/p>

TensorRT-LLM是Nvidia原始深度學(xué)習(xí)軟件庫(kù)的演變，對(duì)LLM推理進(jìn)行了優(yōu)化。它旨在支持 H100，但也可以應(yīng)用于 A100 和 L4 部署。

“[在TensorRT-LLM中，我們]確保我們?yōu)榇笮驼Z(yǔ)言模型提供最佳的張量核心優(yōu)化，”Buck說(shuō)。“這允許人們采用任何大型語(yǔ)言模型并通過(guò)TensorRT-LLM傳遞，以獲得Hopper的變壓器引擎的好處，該引擎使Hopper的FP8計(jì)算能力成為可能。而且在生產(chǎn)工作流程中不會(huì)有任何準(zhǔn)確性損失。

Nvidia 的 Hopper 架構(gòu)引入了變壓器引擎，這是一個(gè)軟件庫(kù)，可智能地管理訓(xùn)練和推理工作負(fù)載的精度，以實(shí)現(xiàn)最佳性能。Buck說(shuō)，變壓器引擎需要對(duì)所涉及的數(shù)學(xué)，統(tǒng)計(jì)數(shù)據(jù)和數(shù)據(jù)有深入的了解，并在Nvidia的編譯器上進(jìn)行大量工作。它有助于在模型投入生產(chǎn)后保持模型的預(yù)測(cè)準(zhǔn)確性，這可能是一個(gè)挑戰(zhàn)。

“你可以很容易地將32位或16位計(jì)算塞進(jìn)FPGA中，但你可能會(huì)得到錯(cuò)誤的答案，因?yàn)樗鼪](méi)有你想要的生產(chǎn)級(jí)精度，”巴克說(shuō)。“深思熟慮和謹(jǐn)慎地做到這一點(diǎn)，保持規(guī)模和偏差，在某些情況下將計(jì)算保持在只有8位的范圍內(nèi) - 為模型的某些部分保留FP16 - 這是Nvidia已經(jīng)努力了一段時(shí)間的事情。

TensorRT-LLM還包括一個(gè)稱為動(dòng)態(tài)批處理的新功能。

Buck解釋說(shuō)，LLM工作負(fù)載，甚至是同一模型的推理工作負(fù)載，都是多種多樣的。LLM從情緒分析等更簡(jiǎn)單的用例開始，但今天的LLM可能正在回答問(wèn)題，閱讀長(zhǎng)文本并總結(jié)它們，或者為電子郵件，文章，演示文稿等生成長(zhǎng)文本或短文本。為L(zhǎng)LM推理服務(wù)的數(shù)據(jù)中心也可以為許多不同的用戶提供許多不同的服務(wù)。

與現(xiàn)有的AI工作負(fù)載相比，現(xiàn)有的AI工作負(fù)載在大小上更有可能相似，因此易于批處理，Buck表示，針對(duì)同一模型的LLM查詢?cè)诖笮》矫婵赡軙?huì)相差幾個(gè)數(shù)量級(jí)，從需要幾毫秒才能完成的查詢到需要幾秒鐘的查詢。模型也可以堆疊，使事情變得更加復(fù)雜。

“我們的標(biāo)準(zhǔn)批處理方法總是等待最長(zhǎng)的查詢完成，”他說(shuō)?！皥D像查詢大致花費(fèi)相同的時(shí)間——從效率的角度來(lái)看，這不是問(wèn)題，而且查詢可以填充，所以沒(méi)什么大不了的。

借助新的動(dòng)態(tài)批處理功能，查詢完成后，查詢可以停用，軟件可以插入另一個(gè)查詢，而較長(zhǎng)的查詢?nèi)栽谶M(jìn)行中。這有助于提高具有不同查詢長(zhǎng)度的 LLM 的 GPU 利用率。

“坦率地說(shuō)，結(jié)果甚至讓我感到驚訝，”巴克說(shuō)?！八笻opper的性能翻了一番。Hopper 是一個(gè)非常強(qiáng)大的 GPU，它可以在同一個(gè) GPU 中并行處理大量查詢，但如果沒(méi)有動(dòng)態(tài)批處理，如果你給它多樣化的查詢，它會(huì)等待最長(zhǎng)的查詢，而不會(huì)被充分利用。

TensorRT-LLM是開源的，以及Nvidia的所有LLM工作，包括許多LLM模型，如GPT，Bloom和Falcon，這些模型已經(jīng)通過(guò)內(nèi)核融合，更快的注意力，多頭注意力等技術(shù)進(jìn)行了優(yōu)化。所有這些操作的內(nèi)核都作為TensorRT-LLM的一部分開源。

“這使得對(duì)性能感興趣的研究人員有一個(gè)起點(diǎn)，使其更快，”巴克說(shuō)?！拔覀兊目蛻艉陀脩艉苄蕾p，如果他們有一個(gè)想要部署的特定想法，他們可以針對(duì)他們的用例進(jìn)一步優(yōu)化一些東西。

創(chuàng)新來(lái)自學(xué)術(shù)界，也來(lái)自Meta、Microsoft和谷歌等公司。雖然 Nvidia 與他們合作優(yōu)化推理，雖然優(yōu)化可能會(huì)成為學(xué)術(shù)論文，但“世界沒(méi)有一個(gè)好地方去獲得這些優(yōu)化，而且 Nvidia 工程師所做的工作沒(méi)有得到一個(gè)可以幫助世界其他地方的地方分享，“巴克說(shuō)。

Buck補(bǔ)充說(shuō)，TensorRT-LLM的性能提升在下一輪MLPerf推理分?jǐn)?shù)中應(yīng)該是顯而易見的，該分?jǐn)?shù)將于明年春天到期。

審核編輯：彭菁