龙图腾网&IPTOP
- 微信扫码登录
- 账号密码登录
- 短信登录/注册
买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！
申请/专利权人：南京砺算科技有限公司;砺算科技(上海)有限公司
摘要：本发明公开了一种基于GPU的隐式GEMM卷积计算方法、装置、设备及介质。该方法包括：接收用户输入的隐式GEMM卷积计算模式，并通过编译器生成与计算模式匹配的指令序列；根据指令序列中的特征加载指令从全局内存中读取三维特征图的第一分块并加载到第一寄存器中；根据指令序列中的常数加载指令从全局内存中读取三维卷积核的第二分块并加载到第二寄存器中；根据指令序列中的DP4A指令对第一分块和第二分块进行多路并行计算，第一分块或第二分块为二维32*32的数据。通过专用的数据加载指令对分块进行加载，所加载的分块较大，并且仅涉及到了从全局内存向寄存器的加载，而不局限于共享内存，提高了数据复用率和执行效率。
主权项：1.一种基于GPU的隐式GEMM卷积计算方法，其特征在于，包括：接收用户输入的隐式GEMM卷积计算模式，并通过编译器生成与所述计算模式匹配的指令序列，其中，所述计算模式包括第一模式或第二模式，所述第一模式所对应的卷积输出通道为32的倍数，所述第二模式所对应的卷积输出通道不是32的倍数；根据所述指令序列中的特征加载指令从全局内存中读取三维特征图的第一分块并加载到第一寄存器中；根据所述指令序列中的常数加载指令从全局内存中读取三维卷积核的第二分块并加载到第二寄存器中；根据所述指令序列中的DPA4指令对所述第一分块和所述第二分块进行多路并行计算，其中，所述第一分块或所述第二分块为二维32*32的数据；所述根据所述指令序列中的DP4A指令对所述第一分块和所述第二分块进行多路并行计算，包括：将所述指令序列中的DP4A指令发送给GPU的数据计算单元；同时运行三路DP4A指令；针对每条DP4A指令将所述第一寄存器中第一分块中的指定线程和所述第二寄存器中第二分块的指定线程进行点乘；并将每条DP4A的点乘结果存放到第三寄存器中。
全文数据：
权利要求：
百度查询： 南京砺算科技有限公司 砺算科技(上海)有限公司 一种基于GPU的隐式GEMM卷积计算方法、装置、设备及介质
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息，力求客观、公正，但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解，仅供参考使用，不能作为本公司承担任何法律责任的依据或者凭证。
主营中国专利交易买卖与专利转让许可,高校科技成果推广与科技成果转化,知识产权运营管理与平台开发,科技人才专家库与科技猎头等科技服务
开放平台
担保交易
惠及多方
会员特惠
专属平台
适合多方
数据共享
功能齐全
皖公网安备 34010402703815号