恭喜西安电子科技大学广州研究院;西安电子科技大学何鑫获国家专利权
买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
龙图腾网恭喜西安电子科技大学广州研究院;西安电子科技大学申请的专利基于密集张量核心的高性能SpMM内核实现方法获国家发明授权专利权,本发明授权专利权由国家知识产权局授予,授权公告号为:CN120045827B 。
龙图腾网通过国家知识产权局官网在2025-07-04发布的发明授权授权公告中获悉:该发明授权的专利申请号/专利号为:202510526626.4,技术领域涉及:G06F17/16;该发明授权基于密集张量核心的高性能SpMM内核实现方法是由何鑫;卜伟锋;郭清华;蔡玉树;盛凯设计研发完成,并于2025-04-25向国家知识产权局提交的专利申请。
本基于密集张量核心的高性能SpMM内核实现方法在说明书摘要公布了:本申请涉及稀疏矩阵乘法技术领域,公开了一种基于密集张量核心的高性能SpMM内核实现方法。方法包括:将稀疏矩阵采用BlockNM稀疏格式进行存储,并将稀疏矩阵中的元数据预取至GPU的共享内存中;将稀疏矩阵和稠密矩阵异步复制至共享内存,设置四阶段流水线调度策略,控制不同迭代过程的SpMM操作;将共享内存中的稀疏矩阵和稠密矩阵加载至寄存器中,为访问共享内存的每个线程设置偏移量,基于偏移量使得每个线程访问共享内存中的不同bank;在寄存器中执行矩阵乘加操作获得计算结果,并将计算结果从寄存器暂存至共享内存,再将共享内存中的计算结果写回至全局内存,本申请提高了稀疏矩阵乘法的效率。
本发明授权基于密集张量核心的高性能SpMM内核实现方法在权利要求书中公布了:1.一种基于密集张量核心的高性能SpMM内核实现方法,其特征在于,所述方法包括: 步骤S1:将稀疏矩阵划分为预设大小的矩阵块,每个矩阵块中的非零元素采用BlockNM稀疏格式进行存储,并获得所述非零元素的列索引数据,定义为元数据,所述元数据通过第一指令预取至GPU的共享内存中; 步骤S2:从全局内存中读取稠密矩阵,并将所述稀疏矩阵和所述稠密矩阵通过第二指令异步复制至所述共享内存中,在所述共享内存中所述稠密矩阵采用行主序格式进行存储,在异步复制至所述共享内存后,设置四阶段流水线调度策略,基于所述四阶段流水线调度策略控制不同迭代过程的矩阵乘加操作; 步骤S3:将所述共享内存中的所述稀疏矩阵和所述稠密矩阵分批次加载至寄存器中,在加载的过程中,为访问所述共享内存的每个线程设置偏移量,基于所述偏移量使得每个线程访问所述共享内存中的不同bank; 步骤S4:基于第三指令在所述寄存器中执行矩阵乘加操作,获得计算结果,并将所述计算结果从所述寄存器暂存至所述共享内存中,基于第四指令将共享内存中的计算结果写回至全局内存中。
如需购买、转让、实施、许可或投资类似专利技术,可联系本专利的申请人或专利权人西安电子科技大学广州研究院;西安电子科技大学,其通讯地址为:510555 广东省广州市黄埔区中新广州知识城知明路83号;或者联系龙图腾网官方客服,联系龙图腾网可拨打电话0551-65771310或微信搜索“龙图腾网”。
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。