支持动态形状的计算的方法、装置、介质和程序产品

摘要文本

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