雖然近年來各大EDA公司都在積極的將AI引入到自己的芯片設計工具當中。但是早在2020年,，谷歌就發(fā)布了題為《Chip Placement with Deep Reinforcement Learning》預印本論文，介紹了其設計芯片布局的新型強化學習方法,。隨后在2021年,，谷歌又在 Nature上發(fā)表了論文并將其開源了。
近日,，谷歌詳細介紹了其用于芯片設計布局的強化學習方法,，并將該模型命名為“AlphaChip” ，據(jù)稱AlphaChip有望大大加快芯片布局規(guī)劃的設計,，并使它們在性能,、功耗和面積方面更加優(yōu)化。目前AlphaChip已發(fā)布在Github上與公眾共享,，同時谷歌還開放了一個在 20 個 TPU 模塊上預訓練的檢查點,。據(jù)介紹，AlphaChip在設計谷歌的張量處理單元 （TPU） 方面發(fā)揮了重要作用,，并已被包括聯(lián)發(fā)科（MediaTek）在內(nèi)的其他公司采用,。
谷歌首席科學家 Jeff Dean 表示，開放預訓練 AlphaChip 模型檢查點以后,，外部用戶可以更容易地使用 AlphaChip 來啟動自己的芯片設計,。
從數(shù)個月縮短至數(shù)小時
通常芯片設計布局或平面圖是芯片開發(fā)中時間最長、勞動強度最高的階段,。近年來,，新思科技（Synopsys）開發(fā)了 AI 輔助芯片設計工具，可以加速開發(fā)并優(yōu)化芯片的布局規(guī)劃。但是,，這些工具非常昂貴,。谷歌希望在一定程度上使這種 AI 輔助芯片設計方法大眾化。
如今,，如果由人類來為 GPU 等復雜芯片設計平面圖大約需要 24 個月,。不太復雜的芯片的平面規(guī)劃可能也至少需要幾個月的時間，而這意味著數(shù)百萬美元的成本,，因為維持一個設計團隊通常需要一大筆費用,。
谷歌表示,，AlphaChip 加快了這一時間表,，可以在短短幾個小時內(nèi)創(chuàng)建芯片布局。此外,，據(jù)說它的設計非常出色,，因為它們優(yōu)化了電源效率和性能。谷歌還展示了一張圖表,，顯示與人類開發(fā)人員相比,，各種版本的 TPU 和 Trillium 的平均線長（wirelength）都有所減少。
△圖中展示了 AlphaChip 在三代 Google 張量處理單元 (TPU) 中的平均線長（wirelength）減少量,，并與 TPU 物理設計團隊生成的位置進行了比較,。
AlphaChip 是如何工作的？
芯片設計并非易事,，部分原因在于計算機芯片由許多相互連接的塊組成,，這些塊具有多層電路元件，所有元件都通過極細的導線連接,。此外,，芯片還有很多復雜且相互交織的設計約束，設計時必須同時滿足所有約束,。由于這些復雜性,，芯片設計師們在 60 多年來一直在努力實現(xiàn)芯片布局規(guī)劃過程的自動化。
與 AlphaGo 和 AlphaZero 類似,，谷歌構(gòu)建時 AlphaChip,，也將芯片的布局規(guī)劃視為一種博弈。
AlphaChip 從空白網(wǎng)格開始,，一次放置一個電路元件,，直到完成所有元件的放置。然后根據(jù)最終布局的質(zhì)量給予獎勵,。谷歌提出了一種新穎的「基于邊」的圖神經(jīng)網(wǎng)絡使 AlphaChip 能夠?qū)W習互連芯片元件之間的關系,，并在整個芯片中進行推廣，讓 AlphaChip 在其設計的每一個布局中不斷進步。
△左圖： AlphaChip 在沒有任何經(jīng)驗的情況下放置開源處理器 Ariane RISC-V CPU的電路元件,；右圖： AlphaChip 在對 20 個 TPU 相關設計進行練習后放置相同的電路元件,。
AlphaChip 還使用強化學習模型，其中代理在預設環(huán)境中采取行動,，觀察結(jié)果,，并從這些經(jīng)驗中學習，以便在未來做出更好的選擇,。在 AlphaChip 的情況下,，系統(tǒng)將 chip floorplanning 視為一種游戲，一次將一個 circuit 元件放置在空白網(wǎng)格上,。該系統(tǒng)隨著解決更多布局而改進,，使用圖形神經(jīng)網(wǎng)絡來理解組件之間的關系。
谷歌TPU及聯(lián)發(fā)科天璣芯片設計均已采用
自 2020 年以來,，AlphaChip 一直被用于設計谷歌自己的 TPU AI 加速器,，這些加速器驅(qū)動著谷歌的許多大規(guī)模 AI 模型和云服務。這些處理器運行基于 Transformer 的模型,，為谷歌的 Gemini 和 Imagen 提供支持,。
為了設計 TPU 布局，AlphaChip 首先在前幾代的各種芯片塊上進行練習,，例如片上和芯片間網(wǎng)絡塊,、內(nèi)存控制器和數(shù)據(jù)傳輸緩沖區(qū)。這個過程稱為預訓練,。然后谷歌在當前的 TPU 塊上運行 AlphaChip 以生成高質(zhì)量的布局,。與之前的方法不同，AlphaChip 解決了更多芯片布局任務實例,，因此變得更好,、更快，就像人類專家所做的那樣,。
可以說AlphaChip 改進了每一代 TPU 的設計,，包括最新的第 6 代 Trillium 芯片，確保了更高的性能和更快的開發(fā),。盡管如此,，谷歌和聯(lián)發(fā)科目前仍都只是依賴 AlphaChip 來制作芯片當中有限的一些區(qū)塊，而人類開發(fā)人員仍然承擔了大部分的設計工作,。
△谷歌最近三代張量處理單元 (TPU)（包括 v5e,、v5p 和 Trillium）中 AlphaChip 設計的芯片塊的數(shù)量
到目前為止，AlphaChip 已被用于開發(fā)各種處理器,，包括谷歌的 TPU 和聯(lián)發(fā)科旗艦級天璣 5G SoC芯片,，這些處理器廣泛用于各種智能手機,。此外，還包括谷歌首款基于 Arm 的通用數(shù)據(jù)中心 CPU—— Axion,。因此,，AlphaChip 能夠在不同類型的處理器中進行泛化。
谷歌表示,，它已經(jīng)在各種芯片模塊上進行了預訓練,，這使得 AlphaChip 能夠在實踐更多設計時生成越來越高效的布局。雖然人類專家可以學習,，而且許多人學得很快,，但機器的學習速度要高出幾個數(shù)量級。
擴展AI在芯片開發(fā)中的應用
谷歌表示,，AlphaChip 的成功激發(fā)了一波新的研究浪潮,，將人工智能用于芯片設計的不同階段。這包括將AI技術擴展到邏輯綜合,、宏選擇和時序優(yōu)化等領域,，Synopsys和Cadence已經(jīng)提供了這些技術,，盡管需要很多錢,。據(jù)谷歌稱，研究人員還在探索如何將 AlphaChip 的方法應用于芯片開發(fā)的更進一步階段,。
“AlphaChip 激發(fā)了芯片設計強化學習的全新研究路線,，跨越了從邏輯綜合到布局規(guī)劃、時序優(yōu)化等的設計流程,，”谷歌的一份聲明中寫道,。
展望未來，谷歌看到了 AlphaChip 徹底改變整個芯片設計生命周期的潛力：從架構(gòu)設計到布局再到制造,，人工智能驅(qū)動的優(yōu)化可能會帶來更快的芯片,、更小（即更便宜）和更節(jié)能的芯片,。雖然目前谷歌的服務器和基于聯(lián)發(fā)科天璣 5G 的智能手機受益于 AlphaChip,，但未來應用程序可能會擴展到幾乎所有領域。
目前AlphaChip 的未來版本已經(jīng)在開發(fā)中,，因此或許未來在AI的驅(qū)動下,，芯片設計將會變得更加的簡單。
