[发明专利]面向SW39000处理器的稠密矩阵乘法的高性能实现方法及装置

权利要求书

1.一种面向SW39000处理器的稠密矩阵乘法的高性能实现方法，其步骤包括：

1)在主内存级别分别对非转置矩阵A、非转置矩阵B与问题矩阵C进行任务分块，产生若干的子矩阵δA_il、子矩阵δB_lj与子矩阵δC_ij，其中非转置矩阵A、非转置矩阵B与问题矩阵C的大小分为m×k、k×n与m×n，子矩阵δA_il、子矩阵δB_lj与子矩阵δC_ij的大小分为b_m×b_k、b_k×b_n与b_m×bx；

2)当子矩阵δC_ij更新时，在从核阵列的LDM级别分别对子矩阵δC_ij、子矩阵δA_il及子矩阵δB_lj进行任务分块，将产生的子块εC_uv、子块εA_uw及子块εB_wv映射至相应的从核，其中0≤u≤T，0≤w≤T，0≤v≤T，参数T根据从核数量设定；

3)将子矩阵δA_il的第t个列块与子矩阵δB_lj的第t个行块相乘，生成矩阵δC_t，并基于子矩阵δC_ij与各矩阵δC_t，得到子矩阵δC_ij的更新结果δC′_ij；

4)基于所述更新结果δC′_ij，更新主内存中的子矩阵δC_ij，以得到问题矩阵C的输出结果。

2.如权利要求1所述的方法，其特征在于，非转置矩阵A、非转置矩阵B与问题矩阵C的数据类型包括：实数单精、实数双精、复数单精或复数双精。

3.如权利要求1所述的方法，其特征在于，获取需更新的子矩阵δC_ij的方法包括：采用N-M-K的循环次序，依次遍历每个子矩阵δC_ij。

4.如权利要求1所述的方法，其特征在于，将子矩阵δC_ij、子矩阵δA_il及子矩阵δB_ij传输到从核阵列的LDM中的方法包括：使用DMA数据传输方法。

5.如权利要求1所述的方法，其特征在于，通过以下步骤生成矩阵δC_t：

1)根据子矩阵δA_il列块中子块εA_uw与子矩阵δB_lj行块中的子块εB_wv，获取子块εC_uv所在的从核；

2)子块εC_uv所在的从核，获取子块εA_uw与子块εB_wv；

3)在寄存器级别，将子块εC_uv、子块εA_uw与子块εB_wv分别进行任务划分，得到相应的计算子块；

4)基于子块εA_uw与子块εB_wv的计算子块，利用浮点向量乘加指令与循环展开的方式，并以子块εC_uv的计算子块为单位进行更新，得到子块εC_uv的更新结果；

5)利用各子块εC_uv的更新结果，生成矩阵δC_t。

6.如权利要求5所述的方法，其特征在于，当子块εA_uw与子块εB_wv不在子块εC_uv所在的从核时，该从核基于RMA通信方法，获取子块εA_uw与子块εB_wv。