[发明专利]基于GEMM处理图像数据的方法、装置、设备和介质

权利要求书

1.一种基于通用矩阵乘法处理图像数据的方法，包括：

获取汇编指令集，所述汇编指令集指定针对图像数据的预定操作序列；

对所述汇编指令集进行分析，得到指令集分析结果；

根据所述指令集分析结果分配用于执行所述预定操作序列的寄存器；以及

读取所述图像数据，并利用所分配的寄存器对所述图像数据执行所述预定操作序列，

其中，所述汇编指令集包括自定义的汇编指令，所述自定义的汇编指令包括汇编字符串和操作数约束条件，

其中，对所述汇编指令集进行分析包括：解析所述汇编字符串中的操作符和操作数并解析与操作数对应的操作数约束条件，以确定寄存器分配方式。

2.如权利要求1所述的方法，其中，所述自定义的汇编指令位于所述预定操作序列的每一操作的起始位置。

3.如权利要求2所述的方法，其中，根据所述指令集分析结果分配用于执行所述预定操作序列的寄存器包括：

在汇编字符串中的操作符的类型是标签的情况下，不为该操作符分配寄存器；

在操作数约束条件中包括预定义的寄存器ID和保留关键字的情况下，按照预定义的寄存器ID为保留关键字分配所需数量的寄存器；

在汇编字符串中的操作数是非数字的情况下，从已注册的寄存器列表中查找寄存器ID；

在汇编字符串中的操作数是数字的情况下，检查操作数的地址，在已注册的寄存器列表里查找与所述地址对应的寄存器ID，如果通过查找确定该地址已被注册，则不为该操作数分配新的寄存器，否则，按照操作数所需寄存器个数分配相应的寄存器。

4.如权利要求1至3中的任一项所述的方法，其中，利用所分配的寄存器对所述图像数据执行所述预定操作序列包括：

将图像数据的第一矩阵块和第二矩阵块加载到内存中；

将加载到内存中的所述图像数据的第一矩阵块和第二矩阵块存储到缓存中；

对存储到缓存中的所述图像数据的第一矩阵块和第二矩阵块进行逻辑运算操作；

将通过逻辑运算操作获得的结果矩阵存回内存中。

5.一种基于通用矩阵乘法处理图像数据的装置，包括：

获取模块，被配置为获取汇编指令集，所述汇编指令集指定针对图像数据的预定操作序列；

分析模块，被配置为对所述汇编指令集进行分析，得到指令集分析结果；

分配模块，被配置为根据所述指令集分析结果分配用于执行所述预定操作序列的寄存器；以及

执行模块，被配置为读取所述图像数据，并利用所分配的寄存器对所述图像数据执行所述预定操作序列，

其中，所述汇编指令集包括自定义的汇编指令，所述自定义的汇编指令包括汇编字符串和操作数约束条件，

其中，对所述汇编指令集进行分析包括：解析所述汇编字符串中的操作符和操作数并解析与操作数对应的操作数约束条件，以确定寄存器分配方式。

6.如权利要求5所述的装置，其中，所述自定义的汇编指令位于所述预定操作序列的每一操作的起始位置。

7.如权利要求6所述的装置，其中，根据所述指令集分析结果分配用于执行所述预定操作序列的寄存器包括：

在汇编字符串中的操作符的类型是标签的情况下，不为该操作符分配寄存器；

在操作数约束条件中包括预定义的寄存器ID和保留关键字的情况下，按照预定义的寄存器ID为保留关键字分配所需数量的寄存器；

在汇编字符串中的操作数是非数字的情况下，从已注册的寄存器列表中查找寄存器ID；

在汇编字符串中的操作数是数字的情况下，检查操作数的地址，在已注册的寄存器列表里查找与所述地址对应的寄存器ID，如果通过查找确定该地址已被注册，则不为该操作数分配新的寄存器，否则，按照操作数所需寄存器个数分配相应的寄存器。