[發明專利]一種芯片控制方法、裝置、人工智能芯片及終端設備

在審

• 申請號: 202011013518.0
• 申請日: 2020-09-23
• 公開（公告）號: CN112214315A
• 公開（公告）日: 2021-01-12
• 發明（設計）人: 王和國;黎立煌;蔣文;張丹
• 申請（專利權）人: 深圳云天勵飛技術股份有限公司
• 主分類號: G06F9/50
• 分類號: G06F9/50
• 代理公司: 深圳中一聯合知識產權代理有限公司 44414
• 代理人: 劉永康

本申請屬于人工智能芯片技術領域，尤其涉及一種芯片控制方法、裝置、人工智能芯片及終端設備。所述方法應用于人工智能芯片中，所述方法包括：獲取待處理的神經網絡模型計算任務；將所述神經網絡模型計算任務按照執行順序依次分解為第一子任務、第二子任務和第三子任務；執行所述第一子任務；在所述第一子任務的執行過程中，逐步提升芯片功耗，直至達到預設的峰值功耗；執行所述第二子任務；在所述第二子任務的執行過程中，保持所述峰值功耗；執行所述第三子任務；在所述第三子任務的執行過程中，逐步降低芯片功耗，直至所述神經網絡模型計算任務執行結束。通過本申請，使得人工智能芯片在啟動工作及結束工作的時候均可正常工作。

技術領域

本申請屬于人工智能芯片技術領域，尤其涉及一種芯片控制方法、裝置、人工智能芯片及終端設備。

背景技術

隨著人工智能技術的不斷發展，人工智能芯片的算力需求呈指數級別的增長，算力大意味著功耗也隨之增大，當人工智能芯片從待機狀態進入全速工作時或者從全速工作退出至待機狀態時，功耗變化非常之大，也即工作電流抖動非常之大，但是電源供電芯片對電流的抖動有一定的要求，如圖1所示，如果抖動超過了該要求，則會發生供電電壓不穩定的現象，也即供電電壓出現較大波動，如果波動太大超出人工智能芯片的正常工作電壓，則可能導致人工智能芯片無法正常工作。

發明內容

有鑒于此，本申請實施例提供了一種芯片控制方法、裝置、人工智能芯片及終端設備，以解決大算力的人工智能芯片在啟動工作和結束工作的時候電流抖動大，以至于芯片無法正常工作的問題。

本申請實施例的第一方面提供了一種芯片控制方法，應用于人工智能芯片中，可以包括：

獲取待處理的神經網絡模型計算任務；

將所述神經網絡模型計算任務按照執行順序依次分解為第一子任務、第二子任務和第三子任務；

執行所述第一子任務；在所述第一子任務的執行過程中，逐步提升芯片功耗，直至達到預設的峰值功耗；

執行所述第二子任務；在所述第二子任務的執行過程中，保持所述峰值功耗；

執行所述第三子任務；在所述第三子任務的執行過程中，逐步降低芯片功耗，直至所述神經網絡模型計算任務執行結束。

在本申請實施例的一種具體實現中，所述在所述第一子任務的執行過程中，逐步提升芯片功耗，包括：

將所述人工智能芯片中的各個計算元件劃分為N個計算元件群組，N為正整數；

將所述第一子任務按照執行順序依次分解為N個功耗提升階段；

在第n個功耗提升階段，從尚未啟動的計算元件群組中選取一個計算元件群組作為待啟動群組，并啟動所述待啟動群組，1≤n≤N；

所述在所述第三子任務的執行過程中，逐步降低芯片功耗，包括：

將所述第三子任務按照執行順序依次分解為N個功耗降低階段；

在第n個功耗降低階段，從已啟動的計算元件群組中選取一個計算元件群組作為待關閉群組，并關閉所述待關閉群組。

進一步地，所述將所述人工智能芯片中的各個計算元件劃分為N個計算元件群組，包括：

確定所述人工智能芯片的功耗波動閾值；

根據所述功耗波動閾值將所述人工智能芯片中的各個計算元件劃分為N個計算元件群組，其中，每個計算元件群組的功耗小于所述功耗波動閾值。

進一步地，所述從尚未啟動的計算元件群組中選取一個計算元件群組作為待啟動群組，包括：

分別確定各個計算元件群組的工作優先級；