买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:上海燧原科技股份有限公司
摘要:本发明公开了一种二维卷积算子切分方法、装置、设备及存储介质,包括:获取AI处理器的当前运行参数,根据当前运行参数确定AI处理器的总算力以及最大计算密度,并根据总算力以及最大计算密度构建Roofline模型;根据预先构建的二维卷积算子切分尺寸与AI处理器运行参数之间的数值约束条件,确定二维卷积算子对应的多种备选切分方式;确定与每种备选切分方式对应的模拟数据流,根据模拟数据流以及Roofline模型,确定每种备选切分方式对应的性能数据,并根据性能数据筛选二维卷积算子对应的目标切分方式。本发明实施例的技术方案可以提高切分方式的灵活性以及AI处理器的计算性能。
主权项:1.一种二维卷积算子切分方法,其特征在于,所述方法包括:获取AI处理器的当前运行参数,根据所述当前运行参数确定AI处理器的总算力以及最大计算密度,并根据所述总算力以及最大计算密度构建Roofline模型;根据预先构建的二维卷积算子切分尺寸与AI处理器运行参数之间的数值约束条件,确定二维卷积算子对应的多种备选切分方式;确定与每种备选切分方式对应的模拟数据流,根据所述模拟数据流以及Roofline模型,确定每种备选切分方式对应的性能数据,并根据所述性能数据筛选二维卷积算子对应的目标切分方式。
全文数据:
权利要求:
百度查询: 上海燧原科技股份有限公司 一种二维卷积算子切分方法、装置、设备及存储介质
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。