[发明专利]AI芯片的数据处理方法、装置和计算机设备

权利要求书

1.一种AI芯片的数据处理方法，其特征在于，包括：

在开始对待处理数据进行处理时，确定用于对所述待处理数据进行处理的目标AI模型；

在AI芯片中匹配与所述目标AI模型对应的数据流网络和所述数据流网络的数据流向；

基于匹配了所述数据流网络和所述数据流向的AI芯片对所述待处理数据进行处理。

2.如权利要求1所述的方法，其特征在于，所述在AI芯片中匹配与所述目标AI模型对应的数据流网络和所述数据流网络的数据流向，包括：

确定所述目标AI模型对应的目标流图；

确定所述目标流图中的多个计算节点和所述多个计算节点的计算顺序；

在所述AI芯片中匹配每个计算节点对应的目标计算模块，得到多个目标计算模块；

将所述多个目标计算模块的连接结果作为所述数据流网络；

基于所述多个计算节点的计算顺序确定数据流向。

3.如权利要求2所述的方法，其特征在于，所述基于所述多个计算节点的计算顺序确定数据流向，包括：

基于所述多个计算节点的计算顺序确定所述多个目标计算模块的计算顺序；

基于所述多个目标计算模块的计算顺序作为数据流向。

4.如权利要求2所述的方法，其特征在于，相邻的两个第一子目标计算模块之间设置有第一存储模块，所述第一存储模块用于在上一第一子目标计算模块计算完成，且下一第一子目标计算模块计算未完成时存储所述上一第一子目标计算模块的计算结果，当所述下一第一子目标计算模块计算完成时将所述上一第一子目标计算模块的计算结果传输给所述下一第一子目标计算模块。

5.如权利要求1所述的方法，其特征在于，AI芯片的每两个相邻的计算模块之间设置有数据流交换机，所述基于匹配了所述数据流网络和所述数据流向的AI芯片对所述待处理数据进行处理，包括：

确定所述数据流网络中的目标数据流交换机；

控制所述目标数据流交换机保持开启状态，以使所述待处理数据基于所述开启状态在所述数据流网络中流动，并在流动过程中被所述目标计算模块处理。

6.如权利要求2所述的方法，其特征在于，至少两个第二子目标计算模块通过一个第二存储模块接收所述待处理数据，所述方法还包括：

将所述待处理数据中至少两个第二子目标计算模块所需的相同数据，在所述第二存储模块中复制得到至少两个相同数据，分别传输给所述至少两个第二子目标计算模块。

7.如权利要求1所述的方法，其特征在于，所述目标AI模型包括第一AI模型和第二AI模型，所述在AI芯片中匹配与所述目标AI模型对应的数据流网络和数据流向，包括：

当所述目标AI模型为第一AI模型时，则在AI芯片中匹配与所述第一AI模型对应的数据流网络和数据流向；

当所述目标AI模型为第二AI模型时，则在AI芯片中匹配与所述第二AI模型对应的数据流网络和数据流向。

8.一种AI芯片的数据处理装置，其特征在于，包括：

AI模型确定模块，用于在开始对待处理数据进行处理时，确定用于对所述待处理数据进行处理的目标AI模型；

数据流网络匹配模块，用于在AI芯片中匹配与所述目标AI模型对应的数据流网络和所述数据流网络的数据流向；

数据处理模块，用于基于匹配了所述数据流网络和所述数据流向的AI芯片对所述待处理数据进行处理。

9.如权利要求8所述的装置，其特征在于，所述数据流网络匹配模块包括：

流图确定单元，用于确定所述目标AI模型对应的目标流图；计算节点确定单元，用于确定所述目标流图中的多个计算节点和所述多个计算节点的计算顺序；

计算模块匹配单元，用于在所述AI芯片中匹配每个计算节点对应的目标计算模块，得到多个目标计算模块；

数据流网络确定单元，用于将所述多个目标计算模块的连接结果作为所述数据流网络；数据流向确定单元，用于基于所述多个计算节点的计算顺序确定数据流向。

10.一种计算机设备，其特征在于，包括：

一个或多个处理器；

存储装置，用于存储一个或多个计算机程序；

当所述一个或多个计算机程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如权利要求1-7中任一项所述的AI芯片的数据处理方法。