[發明專利]數據處理裝置、數據處理方法及相關產品

在審

• 申請號: 202111131270.2
• 申請日: 2021-09-26
• 公開（公告）號: CN113837921A
• 公開（公告）日: 2021-12-24
• 發明（設計）人: 不公告發明人
• 申請（專利權）人: 安徽寒武紀信息科技有限公司
• 主分類號: G06T1/20
• 分類號: G06T1/20;G06N3/08;G06N3/063;G06N3/04
• 代理公司: 北京維昊知識產權代理事務所(普通合伙) 11804
• 代理人: 陳姍姍

1.一種數據處理裝置，包括控制電路、第一存儲電路和第二存儲電路，其中：

所述第一存儲電路用于存儲處理前的第一數據；

所述第二存儲電路用于存儲處理后的第二數據；以及

所述控制電路用于：

根據所述第一數據的co維度的大小，確定優選對齊值；

根據所述優選對齊值確定所述co維度的相鄰維度的拆補分配；以及

根據所述拆補分配來配置并執行分塊指令，以將按照第一維度存儲順序存儲在第一存儲電路上的第一數據轉換為按照第二維度存儲順序存儲在第二存儲電路上的第二數據，其中所述第一數據是多維數據，其多維形狀為：

[高維ho]*[中維wo]*[co維度]*[多維混合]

其中所述多維混合至少包括以下多項的各種組合：co、高維ho、低維ho、高維wo、低維wo；

所述第二數據是三維數據，其三維形狀為：

[ho*wo*co]

其中co表示第二數據的最低存儲維度，wo表示第二數據的次低存儲維度，ho表示第二數據的最高存儲維度。

2.根據權利要求1所述的數據處理裝置，其中所述控制電路進一步用于當所述優選對齊值P小于所述分塊指令的基準對齊值M時，按如下確定所述co維度的相鄰維度的拆補分配：

當M/P≤4時，Ws1＝M/P，Ws2＝1；以及

當M/P4時，Ws1＝4，Ws2＝M/4P，

其中Ws1表示從所述co維度的相鄰低維度側拆分Ws1倍的數據補至所述co維度，Ws2表示從所述co維度的相鄰高維度側拆分Ws2倍的數據補至所述co維度。

3.根據權利要求2所述的數據處理裝置，其中所述控制電路進一步用于按如下規則根據所述第一數據的co維度大小確定所述優選對齊值P：

當0co≤4B時，P＝4B；

當2ⁿ*4Bco≤2ⁿ⁺¹*4B，時，P＝2ⁿ⁺¹*4B；以及

當時，P＝M。

4.根據權利要求2-3任一所述的數據處理裝置，其中所述控制電路進一步用于：

根據所述拆補分配，將所述第一數據由所述多維形狀視為分塊處理前的三維等效形狀：

([高維ho]*[中維wo/Ws2])*(Ws2*co維度*Ws1)*(拆補后多維混合)

其中最高維為[高維ho]*[中維wo/Ws2]，次高維為Ws2*co維度*Ws1，最低維為拆補后多維混合，所述拆補后多維混合中的低維wo變為[低維wo/Ws1]；以及

將所述第二數據由所述三維形狀視為分塊處理后的三維等效形狀：

5.根據權利要求4所述的數據處理裝置，其中所述控制電路進一步用于：

對所述三維等效形狀的第一數據按循環執行分塊指令，所述循環包括內層co維度循環、中層wo維度循環和外層ho維度循環。

6.根據權利要求5所述的數據處理裝置，其中所述內層co維度循環包括：

將所述co維度作為一個數據塊進行處理，內層co維度循環次數repeat_co＝1。

7.根據權利要求6所述的數據處理裝置，其中所述中層wo維度循環包括：

按照wo維度每1個數據為一次數據塊，重復repeat_wo次，其中repeat_wo＝所述中維wo/Ws2。

8.根據權利要求7所述的數據處理裝置，其中所述外層ho維度循環包括：

按照ho維度每1個數據為一次數據塊，重復repeat_ho次，其中repeat_ho＝所述最高維中的高維ho。

9.根據權利要求8所述的數據處理裝置，其中所述控制電路進一步用于按如下配置所述分塊指令：

設置分塊指令的后配表，所述后配表用于將所述分塊指令的輸出張量的內層最低維度數據按照所述后配表的指示進行重排。