[發明專利]運算方法、裝置及相關產品

有效

• 申請號: 201811544251.0
• 申請日: 2018-12-17
• 公開（公告）號: CN109711539B
• 公開（公告）日: 2020-05-29
• 發明（設計）人: 不公告發明人
• 申請（專利權）人: 中科寒武紀科技股份有限公司
• 主分類號: G06N3/063
• 分類號: G06N3/063;G06F3/06
• 代理公司: 北京林達劉知識產權代理事務所(普通合伙) 11277
• 代理人: 劉新宇

1.一種數據處理方法，其特征在于，所述方法應用于第一處理器，所述方法包括：

通過第一存儲模塊接收并存儲輸入數據的第一存儲空間的第一地址，所述第一存儲空間為所述輸入數據在第二處理器中的存儲空間；

對人工智能算子進行編譯，所述人工智能算子是從預設的人工智能算子庫中調用的算子；

在所述人工智能算子編譯完成時，根據輸入數據的類型、輸出數據的類型以及人工智能算子的類型，通過所述第一存儲模塊為所述輸入數據在第一處理器中分配第二存儲空間；

根據所述第一地址，將所述輸入數據復制到所述第二存儲空間；

根據所述第二存儲空間的第二地址，通過所述人工智能算子對所述輸入數據執行相應的人工智能運算，得到輸出數據，

所述方法還包括：在將所述輸入數據復制到所述第二存儲空間后，釋放所述第一存儲模塊中的第一地址。

2.根據權利要求1所述的方法，其特征在于，所述方法還包括：

在人工智能算子編譯完成時，通過所述第一存儲模塊為輸出數據在第一處理器中分配第三存儲空間，并在所述第一存儲模塊中存儲所述第三存儲空間的第三地址。

3.根據權利要求2所述的方法，其特征在于，所述方法還包括：

在得到輸出數據時，根據所述第三地址，將所述輸出數據存儲到所述第三存儲空間。

4.一種數據處理方法，其特征在于，所述方法應用于第二處理器，所述方法包括：

在接收到輸入數據時，通過第二存儲模塊為所述輸入數據在第二處理器中分配第一存儲空間，并在所述第二存儲模塊中存儲所述第一存儲空間的第一地址；

將接收到的輸入數據存儲到所述第一存儲空間；

將所述第一地址傳遞到第一處理器中的第一存儲模塊，以使所述第一處理器在人工智能算子編譯完成時存儲所述輸入數據，并對所述輸入數據執行相應的人工智能運算，

其中，所述方法還包括：在所述第一存儲模塊中的第一地址被釋放后，釋放所述第二存儲模塊中的第一地址，以釋放所述第一存儲空間。

5.根據權利要求4所述的方法，其特征在于，將所述第一地址傳遞到第一處理器中的第一存儲模塊，包括：

通過中間存儲模塊將所述第一地址傳遞到所述第一存儲模塊。

6.一種數據處理裝置，其特征在于，所述裝置應用于第一處理器，所述裝置包括：

數據接收模塊，用于通過第一存儲模塊接收并存儲輸入數據的第一存儲空間的第一地址，所述第一存儲空間為所述輸入數據在第二處理器中的存儲空間；

算子編譯模塊，用于對人工智能算子進行編譯，所述人工智能算子是從預設的人工智能算子庫中調用的算子；

第一分配模塊，用于在所述人工智能算子編譯完成時，根據輸入數據的類型、輸出數據的類型以及人工智能算子的類型，通過所述第一存儲模塊為所述輸入數據在第一處理器中分配第二存儲空間；

復制模塊，用于根據所述第一地址，將所述輸入數據復制到所述第二存儲空間；

數據運算模塊，用于根據所述第二存儲空間的第二地址，通過所述人工智能算子對所述輸入數據執行相應的人工智能運算，得到輸出數據，

所述裝置還包括：第一釋放模塊，用于在將所述輸入數據復制到所述第二存儲空間后，釋放所述第一存儲模塊中的第一地址。

7.根據權利要求6所述的裝置，其特征在于，所述裝置還包括：

第二分配模塊，用于在人工智能算子編譯完成時，通過所述第一存儲模塊為輸出數據在第一處理器中分配第三存儲空間，并在所述第一存儲模塊中存儲所述第三存儲空間的第三地址。

8.根據權利要求7所述的裝置，其特征在于，所述裝置還包括：

第一數據存儲模塊，用于在得到輸出數據時，根據所述第三地址，將所述輸出數據存儲到所述第三存儲空間。