龙图腾网&IPTOP
- 微信扫码登录
- 账号密码登录
- 短信登录/注册
买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！
摘要：本申请实施例提供了一种模型推理优化方法、设备、存储介质及程序产品，涉及人工智能技术领域，该方法包括：在接收到提示内容后，根据提示内容在插槽中对应的目标槽位索引，生成提示内容对应的计算单元索引表，计算单元索引表包括：对提示内容进行多头注意力机制的计算时，计算每个注意力头的计算单元的索引，这样，在模型推理过程中，基于计算单元索引表，可将每个注意力头的计算任务分配到相应的计算单元进行计算，获得计算结果。该方法将控制粒度调整到每个注意力头，这样可以避免计算单元重复计算注意力头的情况，从宏观层面上来说，可以更好地让计算单元并行服务，从而充分利用人工智能芯片的硬件特性，提高读取数据性能和吞吐率。
主权项：1.一种模型推理优化方法，其特征在于，包括：接收提示内容，并获取所述提示内容在插槽中对应的目标槽位索引；根据所述目标槽位索引，生成所述提示内容对应的计算单元索引表，所述计算单元索引表包括：对所述提示内容进行多头注意力机制的计算时，计算每个注意力头的计算单元的索引；为所述提示内容锁定相应的目标物理存储块；所述提示内容包括多个Token；所述目标物理存储块中的每个存储空间，用于保存一个Token关联的键值数据；在模型推理过程中，针对所述多头注意力机制中的多个注意力头，分别执行以下操作：基于所述计算单元索引表中一个注意力头对应的计算单元的索引，将所述一个注意力头的计算任务调度到相应的计算单元；通过所述计算单元对所述提示内容中的所述一个注意力头对应的头数据进行注意力机制计算，获得相应的计算结果，其中，注意力机制计算过程中产生的键值数据缓存在所述目标物理存储块中。
全文数据：
权利要求：
百度查询： 北京壁仞科技开发有限公司 上海壁仞科技股份有限公司 一种模型推理优化方法、设备、存储介质及程序产品
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息，力求客观、公正，但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解，仅供参考使用，不能作为本公司承担任何法律责任的依据或者凭证。
主营中国专利交易买卖与专利转让许可,高校科技成果推广与科技成果转化,知识产权运营管理与平台开发,科技人才专家库与科技猎头等科技服务
开放平台
担保交易
惠及多方
会员特惠
专属平台
适合多方
数据共享
功能齐全
皖公网安备 34010402703815号