申请/专利权人：北京壁仞科技开发有限公司;上海壁仞科技股份有限公司

申请日：2024-01-18

公开（公告）日：2024-04-26

公开（公告）号：CN117609670B

主分类号：G06F17/10

分类号：G06F17/10;G06F17/15;G06F17/16

优先权：

专利状态码：有效-授权

法律状态：2024.04.26#授权;2024.03.15#实质审查的生效;2024.02.27#公开

摘要：本发明涉及一种用于支持动态形状的计算的方法、装置、介质和程序产品。该方法包括：将用户输入的形状拆分至针对单计算核的形状，以便至少基于针对单计算核的形状确定核函数的循环结构；将核函数的循环结构分解为内层循环和外层循环，以使得内层循环匹配缓冲区并且循环展开；以及将核函数的循环结构的循环层拆分为循环块，以便通过复用循环块以使得核函数支持动态维度。本发明能够避免访存和算力的浪费，提高执行性能。

主权项：1.一种用于支持动态形状的计算的方法，其特征在于，所述方法包括：将用户输入的形状拆分至针对单计算核的形状，以便至少基于针对单计算核的形状确定核函数的循环结构；将核函数的循环结构分解为内层循环和外层循环，以使得内层循环匹配缓冲区并且循环展开；将核函数的循环结构的循环层拆分为循环块，以便通过复用循环块以使得核函数支持动态维度，其中，将核函数的循环结构的循环层拆分为循环块，以便通过复用循环块以使得核函数支持动态维度包括：将核函数的各层循环代码拆分为多段分代码，以便对多段分代码所包括的主体分代码进行循环展开；以及配置寄存器所存储的循环次数，以用于动态控制主体分代码的循环次数。

全文数据：

权利要求：

百度查询： 北京壁仞科技开发有限公司;上海壁仞科技股份有限公司 支持动态形状的计算的方法、装置、介质和程序产品

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

阅读全文 双屏查看 官方信息 专利公告 收藏专利 下载PDF 下载WORD