上海壁仞科技股份有限公司请求不公布姓名获国家专利权
买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
龙图腾网获悉上海壁仞科技股份有限公司申请的专利一种注意力机制计算方法、装置、介质、产品及设备获国家发明授权专利权,本发明授权专利权由国家知识产权局授予,授权公告号为:CN121599021B 。
龙图腾网通过国家知识产权局官网在2026-04-14发布的发明授权授权公告中获悉:该发明授权的专利申请号/专利号为:202610114446.X,技术领域涉及:G06N3/063;该发明授权一种注意力机制计算方法、装置、介质、产品及设备是由请求不公布姓名;请求不公布姓名;请求不公布姓名;请求不公布姓名;请求不公布姓名设计研发完成,并于2026-01-28向国家知识产权局提交的专利申请。
本一种注意力机制计算方法、装置、介质、产品及设备在说明书摘要公布了:本发明公开了一种注意力机制计算方法、装置、介质、产品及设备,计算集群中配置M≥4个沿KV序列长度方向并行的CU,每个CU中配置N≥2个沿Q序列长度方向并行的消费者线程束组,消费者线程束组、CU循环加载Q分块、KV分块,经QK计算获得S分块,对S分块进行局部规约和归一化,获得局部规约参数和P分块;经PV计算获得局部QKV分块;对M个CU的局部规约参数、局部QKV分块进行全局规约获得全局规约参数、全局QKV分块;根据全局规约参数对全局QKV分块及上一个循环获得的注意力输出结果进行更新,获得当前的注意力输出结果。本发明能够提高算子对计算资源的使用率,最大化利用硬件资源的计算能力。
本发明授权一种注意力机制计算方法、装置、介质、产品及设备在权利要求书中公布了:1.一种注意力机制计算方法,其特征在于,每个计算集群中配置有用于沿键值序列长度方向并行且循环加载不同键值分块的M个计算单元,每个计算单元中配置有用于沿查询序列长度方向并行且循环加载N个查询分块的N个消费者线程束组,M≥4,N≥2;所述方法由每个计算单元执行,包括: 在每个循环中,针对每个消费者线程束组当前加载的查询分块,将当前加载的查询分块与当前加载的键分块进行矩阵乘法计算,获得注意力分数分块; 对所述注意力分数分块进行局部规约计算和归一化处理,获得局部规约参数和注意力权重分块; 根据M个计算单元获得的局部规约参数进行全局规约计算,获得全局规约参数; 将所述注意力权重分块与当前加载的值分块进行矩阵乘法计算,获得局部注意力输出分块; 根据M个计算单元获得的局部注意力输出分块进行全局规约计算,获得全局注意力输出分块; 根据所述全局规约参数对所述全局注意力输出分块及上一个循环获得的注意力输出结果进行更新,获得当前的注意力输出结果。
如需购买、转让、实施、许可或投资类似专利技术,可联系本专利的申请人或专利权人上海壁仞科技股份有限公司,其通讯地址为:201114 上海市闵行区陈行公路2388号16幢13层1302室;或者联系龙图腾网官方客服,联系龙图腾网可拨打电话0551-65771310或微信搜索“龙图腾网”。
以上内容由龙图腾AI智能生成。
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。

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