[发明专利]面向GPDSP的大规模矩阵乘法计算的方法

摘要

本发明公开了一种面向GPDSP的大规模矩阵乘法计算的方法，其步骤为S1由CPU核在片外DDR存储器为矩阵A、B和C分配存储空间，并生成初始化数据或者从其他数据来源传输本次计算所需要的数据；S2根据GPDSP的体系结构特征确定最佳的分块矩阵乘法所需要的分块大小MB、KB和NB；S3由CPU核根据步骤S2确定的MB、KB、NB值将矩阵A、B和C进行逻辑分块；S4由GPDSP的CPU核调度DSP核计算子块矩阵的乘加计算Cik+＝Aij*Bjk(1≤i≤mi,1≤j≤ki,1≤k≤ni)；S5计算完毕。本发明原理简单、操作方便，能充分利用GPDSP的CPU核通用计算和DSP核向量处理阵列强大并行计算、高带宽向量数据加载能力，显著提高DSP核计算访存比。

主权项

一种面向GPDSP的大规模矩阵乘法计算的方法，其特征在于，步骤为：S1：由GPDSP的CPU核在片外DDR存储器为矩阵A、B和C分配存储空间，并生成初始化数据或者从其他数据来源传输本次计算所需要的数据；S2：根据GPDSP的体系结构特征确定最佳的分块矩阵乘法所需要的分块大小MB、KB和NB，即令矩阵的M、K、N三个维数对应的分块大小分别标记为MB、KB、NB；S3：由GPDSP的CPU核根据步骤S2确定的MB、KB、NB值将矩阵A、B和C进行逻辑分块，令mi＝INT(M/MB)，ki＝INT(K/KB)，ni＝INT(N/NB)，其中INT()表示对括号中的值向上取整；矩阵A的子块标记为Aij，1≤i≤mi,1≤j≤ki，矩阵B的子块标记为Bjk，1≤j≤ki,1≤k≤ni，矩阵C的子块标记为Cik，1≤i≤mi,1≤k≤ni；S4：由GPDSP的CPU核调度DSP核计算子块矩阵的乘加计算：Cik+＝Aij*Bjk，1≤i≤mi,1≤j≤ki,1≤k≤ni；S5：计算完毕；所述步骤S2中NB的确定方法是NB＝p*q；其中，p为DSP核的向量处理阵列计算单元数量，q为每个计算单元的乘加部件MAC数量；所述步骤S2中MB、KB的确定方法为：设条件(1)MB*KB*w≤s3，(2)(MB*KB*w+KB*NB*w)≤s1，按照满足条件(1)(2)取得的最大MB、KB值为所需要的MB、KB值；其中，DSP核的片内向量阵列存储器容量为s1字节，GPDSP的片内共享存储阵列容量为s3字节，每个矩阵元素的数据为w字节。