上海壁仞科技股份有限公司请求不公布姓名获国家专利权
买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
龙图腾网获悉上海壁仞科技股份有限公司申请的专利一种注意力机制计算方法、设备、介质及产品获国家发明授权专利权,本发明授权专利权由国家知识产权局授予,授权公告号为:CN121189402B 。
龙图腾网通过国家知识产权局官网在2026-01-30发布的发明授权授权公告中获悉:该发明授权的专利申请号/专利号为:202511716140.3,技术领域涉及:G06N3/063;该发明授权一种注意力机制计算方法、设备、介质及产品是由请求不公布姓名;请求不公布姓名;请求不公布姓名设计研发完成,并于2025-11-21向国家知识产权局提交的专利申请。
本一种注意力机制计算方法、设备、介质及产品在说明书摘要公布了:本申请公开了一种注意力机制计算方法、设备、介质及产品,方法包括:第一内核控制第二线程束组加载第i个查询矩阵分块;第一内核控制第二线程束组计算第j个注意力分数矩阵分块;第一内核控制第一线程束组和第二线程束组交替利用第j个值矩阵分块的不同子分块,与第j个注意力分数矩阵分块进行注意力机制运算的预设前序阶段,直至得到第i个查询矩阵分块的最后1个注意力结果矩阵分块的不同子分块的中间结果;第二内核控制第三线程束组加载第i个查询矩阵分块的最后1个注意力结果矩阵分块的所有子分块的中间结果,并进行注意力机制运算的预设后序阶段。采用本申请实施例能够为注意力机制的计算提供充足的寄存器资源,从而提高计算效率。
本发明授权一种注意力机制计算方法、设备、介质及产品在权利要求书中公布了:1.一种注意力机制计算方法,其特征在于,包括: 第一内核控制第二线程束组加载第i个查询矩阵分块; 所述第一内核控制所述第二线程束组利用第i个查询矩阵分块与转置后的第j个键矩阵分块进行矩阵乘运算和指数运算,得到第j个注意力分数矩阵分块; 所述第一内核控制第一线程束组和所述第二线程束组交替利用第j个值矩阵分块的不同子分块,与第j个注意力分数矩阵分块进行注意力机制运算的预设前序阶段,分别得到第j个注意力结果矩阵分块的不同子分块的中间结果,令j=j+1,返回至所述第一内核控制所述第二线程束组利用第i个查询矩阵分块与转置后的第j个键矩阵分块进行矩阵乘运算和指数运算的步骤,直至j=J,得到第i个查询矩阵分块的最后1个注意力结果矩阵分块的不同子分块的中间结果;其中,J为预设分块总数量,第j个键矩阵分块和第j个值矩阵分块被预先存储于共享内存中; 第二内核控制第三线程束组加载第i个查询矩阵分块的最后1个注意力结果矩阵分块的所有子分块的中间结果,并进行注意力机制运算的预设后序阶段,得到第i个查询矩阵分块的最后1个注意力结果矩阵分块。
如需购买、转让、实施、许可或投资类似专利技术,可联系本专利的申请人或专利权人上海壁仞科技股份有限公司,其通讯地址为:201114 上海市闵行区陈行公路2388号16幢13层1302室;或者联系龙图腾网官方客服,联系龙图腾网可拨打电话0551-65771310或微信搜索“龙图腾网”。
以上内容由龙图腾AI智能生成。
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。

皖公网安备 34010402703815号
请提出您的宝贵建议,有机会获取IP积分或其他奖励