首页 专利交易 科技果 科技人才 科技服务 国际服务 商标交易 会员权益 IP管家助手 需求市场 关于龙图腾
 /  免费注册
到顶部 到底部
清空 搜索

模型计算方法以及相关装置 

买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!

申请/专利权人:中昊芯英(杭州)科技有限公司

摘要:本发明实施例提供一种模型计算方法以及相关装置,属于人工智能技术领域。该方法包括:根据物理设备的显存容量,将目标机器学习模型的模型参数矩阵划分为多个子矩阵;根据多个子矩阵分别对应的重要性等级,确定当前待进行计算的目标子矩阵;根据物理设备包含的多个计算单元分别对应的实时负载信息,从多个计算单元中确定出目标计算单元;将目标子矩阵加载至物理设备的显存,以使目标计算单元基于显存中加载的目标子矩阵执行目标子矩阵对应的矩阵计算任务。该方法通过对模型参数矩阵进行子矩阵的划分,按照子矩的重要性等级依次进行子矩阵计算,显存中依次加载的子矩阵的数据量小于模型参数矩阵,能够有效解决显存占用高、计算资源不足的问题。

主权项:1.一种模型计算方法,其特征在于,所述方法包括:根据物理设备的显存容量,将目标机器学习模型的模型参数矩阵划分为多个子矩阵;确定所述多个子矩阵分别对应的重要性等级;根据所述多个子矩阵分别对应的重要性等级,确定当前待进行计算的目标子矩阵,所述目标子矩阵为所述多个子矩阵中未进行矩阵计算的子矩阵,所述目标子矩阵的重要性等级高于其他未进行矩阵计算的子矩阵的重要性等级;获取所述物理设备包含的多个计算单元分别对应的实时负载信息;根据所述多个计算单元分别对应的实时负载信息,从所述多个计算单元中确定出目标计算单元,所述目标计算单元当前的负载低于其他计算单元当前的负载;将所述目标子矩阵加载至所述物理设备的显存,以使所述目标计算单元基于所述显存中加载的所述目标子矩阵执行所述目标子矩阵对应的矩阵计算任务。

全文数据:

权利要求:

百度查询: 中昊芯英(杭州)科技有限公司 模型计算方法以及相关装置

免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。