买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
摘要:本披露公开了一种计算图的编译方法及相关产品。方法包括:获取待编译的由第二中间表示所表达的计算图,进行形状的前向推导,根据前向推导和张量数据拆分信息,获得完整的形状信息;利用完整的形状信息,判断是否需要调整张量数据拆分信息;根据判断结果确定符合要求的张量数据拆分信息;基于符合要求的张量数据拆分信息对计算图中的算子进行访存模式推导;基于推导的访存模式,确定计算图的代码逻辑中的循环涉及的指令的地址域相关参数;对所述循环中的指令进行流水排布;根据循环涉及的指令的地址域相关参数以及指令流水排布的结果对计算图进行编译,获得智能处理器可识别的二进制文件。
主权项:1.一种计算图的编译方法,其中,所述方法包括:获取待编译的由第二中间表示所表达的计算图,进行形状的前向推导,根据所述前向推导和张量数据拆分信息,获得完整的形状信息;其中,所述第二中间表示是基于第一中间表示经转换获得,所述第一中间表示是基于至少一种前端语言表示的任务模型统一转换获得的;所述计算图是执行在智能处理器上的任务;利用所述完整的形状信息,判断是否需要调整张量数据拆分信息;根据判断结果确定符合要求的张量数据拆分信息;基于符合要求的张量数据拆分信息对所述计算图中的算子进行访存模式推导;基于推导的访存模式,确定所述计算图的代码逻辑中的循环涉及的指令的地址域相关参数;基于所述智能处理器支持的指令流,对所述循环中的指令进行流水排布;根据所述循环涉及的指令的地址域相关参数以及所述循环中的指令流水排布的结果对所述计算图进行编译,获得所述智能处理器可识别的二进制文件。
全文数据:
权利要求:
百度查询: 上海寒武纪信息科技有限公司 计算图的编译方法及相关产品
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。