[发明专利]一种稀疏矩阵向量乘访存优化的计算方法在审
申请号: | 202210066814.X | 申请日: | 2022-01-20 |
公开(公告)号: | CN114491402A | 公开(公告)日: | 2022-05-13 |
发明(设计)人: | 胡长军;储根深;何远杰;丁哲昭;董玲玉;戴潮虎 | 申请(专利权)人: | 北京科技大学 |
主分类号: | G06F17/16 | 分类号: | G06F17/16;G06F9/50 |
代理公司: | 北京市广友专利事务所有限责任公司 11237 | 代理人: | 张仲波;付忠林 |
地址: | 100083*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明公开了一种稀疏矩阵向量乘访存优化的计算方法,适用于DCU和GPU架构,该方法包括:将原始稀疏矩阵按照固定行数划分为若干块,每块由一个线程块单独完成计算,并为每个线程块在LDS开辟固定大小的空间;计算每一线程块需要计算的轮次数;在一个轮次的计算中,每个线程块中所有线程完成若干次非零元素计算并将结果写入LDS;每个线程块中一个或多个线程对LDS的乘法结果进行求和并将结果保存至寄存器;当完成所有轮次计算后,将寄存器中的结果进行进一步计算,并将结果写回显存。本发明基于原始CSR格式,无需预处理,且能充分利用访存合并特性,实现了较高的访存带宽利用。 | ||
搜索关键词: | 一种 稀疏 矩阵 向量 乘访存 优化 计算方法 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京科技大学,未经北京科技大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202210066814.X/,转载请声明来源钻瓜专利网。