上海寒武纪申请计算图编译方法专利,能够实现对计算图高效编译

上海寒武纪申请计算图编译方法专利,能够实现对计算图高效编译
2024年12月09日 15:55 金融界网站

本文源自:金融界

金融界2024年12月9日消息,国家知识产权局信息显示,上海寒武纪信息科技有限公司申请一项名为“计算图的编译方法及相关产品”的专利,公开号CN 119088397 A,申请日期为2024年8月。

专利摘要显示,本披露公开了一种计算图的编译方法及相关产品。方法包括:获取待编译的由第二中间表示所表达的计算图,进行形状的前向推导,根据前向推导和张量数据拆分信息,获得完整的形状信息;利用完整的形状信息,判断是否需要调整张量数据拆分信息;根据判断结果确定符合要求的张量数据拆分信息;基于符合要求的张量数据拆分信息对计算图中的算子进行访存模式推导;基于推导的访存模式,确定计算图的代码逻辑中的循环涉及的指令的地址域相关参数;对所述循环中的指令进行流水排布;根据循环涉及的指令的地址域相关参数以及指令流水排布的结果对计算图进行编译,获得智能处理器可识别的二进制文件。

财经自媒体联盟更多自媒体作者

新浪首页 语音播报 相关新闻 返回顶部