[發明專利]一種量子隨機電路的多振幅模擬方法和裝置

有效

• 申請號: 202211068173.8
• 申請日: 2022-09-02
• 公開（公告）號: CN115130675B
• 公開（公告）日: 2023-01-24
• 發明（設計）人: 劉鑫;劉勇;宋佳偉;陳垚鍵;史新民;彭達佳;郭楚;付昊桓;吳汶釗;楊雨靈;趙朋朋
• 申請（專利權）人: 之江實驗室;國家超級計算無錫中心
• 主分類號: G06N10/20
• 分類號: G06N10/20;G06N10/40;G06F15/78
• 代理公司: 杭州浙科專利事務所(普通合伙) 33213
• 代理人: 孫孟輝

1.一種量子隨機電路的多振幅模擬方法，其特征在于，包括以下步驟：

步驟一：根據芯片片上存儲的最大存儲容量以及張量網絡收縮路徑中每一步張量的最大維數確定進行算子融合的起始位置至結束位置，具體為：當維數最大的張量所需的存儲空間小于片上存儲的最大存儲容量的三分之一時，則這一步的張量網絡收縮可進行算子融合，即可得到進行算子融合的起始位置至結束位置；

步驟二：在起始位置，分別將片外存儲中的兩個張量讀取到片上存儲并進行相應的轉置，再進行融合，具體包括以下子步驟：

步驟2.1：在起始位置，通過直接存儲器訪問讀取方式將片外存儲中的張量B讀取到片上存儲的本地數據內存ldmC位置，在片上存儲中對張量B進行轉置，將轉置后的結果存放到片上存儲的本地數據內存ldmB位置；

步驟2.2：通過直接存儲器訪問讀取方式將片外存儲中的張量A1讀取到片上存儲的本地數據內存ldmC位置，在片上存儲中對張量A1進行轉置，將轉置后的結果存放到片上存儲的本地數據內存ldmA位置；

步驟2.3：在片上存儲中完成張量A1和張量B的融合，將融合的結果存放到片上存儲的本地數據內存ldmC位置；

步驟三：將融合的結果進行轉置后儲存，用于在下一位置與所讀取后轉置的張量進行融合，具體包括以下子步驟：

步驟3.1：在片上存儲中對融合的結果即存放在片上存儲的本地數據內存ldmC位置的融合結果進行轉置，將轉置后的結果存放到片上存儲的本地數據內存ldmB位置；

步驟3.2：在下一個位置，通過直接存儲器訪問讀取方式從片外存儲中將張量A2讀取到片上存儲的本地數據內存ldmC位置，在片上存儲中對張量A2進行轉置，將轉置后的結果存放到片上存儲的本地數據內存ldmA位置；

步驟3.3：在片上存儲中完成張量A2和張量B的融合，將融合結果存放到片上存儲的本地數據內存ldmC位置；

步驟四：重復上述步驟三，直到算子融合到結束位置，將最終的融合的結果寫回到片外存儲。

2.如權利要求1所述的一種量子隨機電路的多振幅模擬方法，其特征在于，所述張量網絡是由多個張量按照收縮規則構成的網絡模型。

3.如權利要求1所述的一種量子隨機電路的多振幅模擬方法，其特征在于，所述步驟四具體為：重復上述步驟三，直到算子融合到結束位置，將存放在本地數據內存ldmC位置的融合結果通過直接存儲器訪問寫入方式寫回到片外存儲張量C位置。

4.一種量子隨機電路的多振幅模擬裝置，其特征在于，包括一個或多個處理器，用于實現權利要求1-3中任一項所述的一種量子隨機電路的多振幅模擬方法。

5.一種計算機可讀存儲介質，其特征在于，其上存儲有程序，該程序被處理器執行時，實現權利要求1-3中任一項所述的一種量子隨機電路的多振幅模擬方法。