一种基于对等内存访问的GPU合并舒尔补的方法2025

专利内容

(19)国家知识产权局 (12)发明专利申请 (10)申请公布号 CN 117806828 A (43)申请公布日 2024.04.02 (21)申请号 202410088800.7 (22)申请日 2024.01.22 (71)申请人 深圳华大九天科技有限公司 地址 518031 广东省深圳市福田区福保街 道福保社区桃花路与槟榔道交汇处西 北深九科技创业园5号楼1001 (72)发明人 陈婧蕊　吴涛　卓立文　陈芳芳　 程明厚　郑瑞卿　 (74)专利代理机构 北京德崇智捷知识产权代理 有限公司 11467 专利代理师 王金双 (51)Int.Cl. G06F 9/50 (2006.01) G06F 17/16 (2006.01) G06T 1/20 (2006.01) 权利要求书1页 说明书5页 附图3页 (54)发明名称 一种基于对等内存访问的GPU合并舒尔补的 方法 (57)摘要 一种基于对等内存访问的GPU合并舒尔补的 方法，包括：开启GPU设备之间的对等内存访问功 能；对稀疏矩阵进行对角分块处理，在不同GPU上 分解底层矩阵，存储为稀疏格式；对底层矩阵进 行编号，并建立编号到底层矩阵数据存储地址的 映射关系；遍历底层矩阵的舒尔补，根据每个位 置的非零元素数量计算得到第一数组；按上层矩 阵非零元素的存储顺序，对底层矩阵的舒尔补的 所有非零元素进行排序，取出排序后的非零元素 的位置和所在矩阵的编号，得到第二数