[发明专利]用于卷积运算的数据重用方法、运算方法及装置、芯片

权利要求书

1.一种用于卷积运算的数据重用方法，其特征在于，包括：

确定执行卷积运算时的限制条件；其中，所述限制条件包括并行通道数量以及每个数据在片上存储器所分配的内存块数量；

根据所述限制条件，确定每个所述数据对应的每个维度上的划分尺寸；

根据所述划分尺寸，将每个所述数据对应的每个所述维度进行划分，获得每个所述数据对应的每个所述维度的候选子图块集合；

将每个所述维度的所述候选子图块集合中的每一个候选子图块，与其余所述维度的候选子图块集合中的每一个候选子图块进行组合，获得每个所述数据的候选图块集合；

将每个所述候选图块集合分别输入至预设的代价函数，并选取所述代价函数的最小输出值对应的候选图块作为目标图块，并将所述目标图块作为所述数据的后续重用划分方式；其中，所述代价函数用于衡量DMA搬运次数。

2.根据权利要求1所述的数据重用方法，其特征在于，所述将每个所述候选图块集合分别输入至预设的代价函数，包括：

根据每个所述数据所分配的所述内存块数量，从每个所述候选图块集合中剔除无效的候选图块，获得有效候选图块集合；

将每个所述有效候选图块集合分别输入至所述代价函数。

3.根据权利要求2所述的数据重用方法，其特征在于，所述将每个所述有效候选图块集合分别输入至所述代价函数，包括：

根据卷积运算的维度的先后顺序，确定所述有效候选图块集合中每个有效候选图块的有效候选图块组合，获得每个所述数据的有效候选图块组合集合；

将每个所述有效候选图块组合集合分别输入至所述代价函数。

4.根据权利要求1～3任一项所述的数据重用方法，其特征在于，所述根据所述划分尺寸，将每个所述数据对应的每个所述维度进行划分，获得每个所述数据对应的每个所述维度的候选子图块集合，包括：

根据所述划分尺寸，对输出数据对应的输出宽度、输出高度和输出通道数量三个维度进行划分、以及，对权重数据对应的输出宽度、输出高度、输入通道数量和输出通道数量四个维度进行划分，获得所述候选子图块集合。

5.一种用于卷积运算的数据重用装置，其特征在于，包括：

一个确定模块，用于确定执行卷积运算时的限制条件；其中，所述限制条件包括并行通道数量以及每个数据在片上存储器所分配的内存块数量；

一个计算模块，用于根据所述限制条件，确定每个所述数据对应的每个维度上的划分尺寸；

一个划分模块，用于根据所述划分尺寸，将每个所述数据对应的每个所述维度进行划分，获得每个所述数据对应的每个所述维度的候选子图块集合；

一个组合模块，用于将每个所述维度的所述候选子图块集合中的每一个候选子图块，与其余所述维度的候选子图块集合中的每一个候选子图块进行组合，以获得每个所述数据的候选图块集合；

一个选取模块，用于将每个所述候选图块集合分别输入至预设的代价函数，并选取所述代价函数的最小输出值对应的候选图块作为目标图块，并将所述目标图块作为所述数据的后续重用划分方式；其中，所述代价函数用于衡量DMA搬运次数。

6.根据权利要求5所述的数据重用装置，其特征在于，所述选取模块包括一个剔除单元和一个输入单元，所述将每个所述候选图块集合分别输入至预设的代价函数，包括：

所述剔除单元，用于根据每个所述数据所分配的所述内存块数量，从每个所述候选图块集合中剔除无效的候选图块，获得有效候选图块集合；

所述输入单元，用于将每个所述有效候选图块集合分别输入至所述代价函数。

7.根据权利要求6所述的数据重用装置，其特征在于，所述选取模块还包括一个组合单元，所述将每个所述有效候选图块集合分别输入至所述代价函数，包括：

所述组合单元，用于根据卷积运算的维度的先后顺序，确定所述有效候选图块集合中每个有效候选图块的有效候选图块组合，获得每个所述数据的有效候选图块组合集合；

所述输入单元，还用于将每个所述有效候选图块组合集合分别输入至所述代价函数。