[发明专利]一种基于GPU的压缩稀疏矩阵的对称矩阵构造方法

摘要

本发明实施例公开了一种基于GPU的压缩稀疏矩阵的对称矩阵构造方法，包括给定基于CSR存储格式的压缩稀疏矩阵M作为输入矩阵；根据压缩稀疏矩阵M，并行地将CSR存储格式转换为三元组数组T1；并行地将三元组数组T1中每一个三元组及其对称的三元组保存到三元组数组T2并进行并行排序得到三元组数组T3；查找T3中重复数据，构造数组F标记重复存储的元素，并行地删除T3中F标记为1的重复存储的元素，得到无重复项的三元组数组T4；根据三元组数组T4，并行地将其转换成CSR存储格式作为输出矩阵。实施本发明，能有效地改善求解稀疏矩阵的对称矩阵的处理性能，使每一个步骤都具有可并行性，发挥GPU中高效并行处理能力。

主权项

一种基于GPU的压缩稀疏矩阵的对称矩阵构造方法，其特征在于，所述方法包括：步骤S101、给定一个基于CSR存储格式的压缩稀疏矩阵M作为输入矩阵；步骤S102、根据输入的压缩稀疏矩阵M，并行地将CSR存储格式转换为一个三元组数组T1(p,q,v)；其中，p表示元素行号，q表示元素列号，v表示矩阵中(p,q)位置对应的元素值；步骤S103、根据构造的三元组数组T1，对于T1中的每一个三元组(p,q,v)，并行地将(p,q,v)及其对称的三元组(q,p,v)保存到新的三元组数组T2，并对得到的新的三元组数组T2以((p<<32)|q)为排序关键字进行并行排序得到新的三元组数组T3；步骤S104、根据排序后得到的三元组T3，查找三元组T3中重复数据，并构造一个数组F用于标记三元组T3中的重复存储的三元组元素，并行地删除三元组T3中通过数组F标记为1的重复存储的元素，得到无重复项的三元组数组T4；步骤S105、根据得到的无重复项的三元组数组T4，并行地将其转换成CSR存储格式作为输出矩阵M1。