[发明专利]数据计算流图的GPU处理方法及装置

摘要

本发明提供了一种数据计算流图的GPU处理方法及装置，该方法包括：将接收的输入数据存储至为计算流图中的源节点分配的第一显存块；获取该源节点所指向的后继节点的表达式，从该第一显存块读取输入数据，以及根据为计算流图中的源节点分配的显存块为该后继节点分配第二显存块；在已获取到该后继节点的表达式所有需要输入的数据的情况下，将读取的输入数据输入至该表达式进行计算，得到该后继节点的结果，并将该后继节点的结果存储至第二显存块；从第二显存块读取该后继节点的结果，并根据读取的后继节点的结果得到输入数据按计算流图处理的结果并输出处理结果。通过上述方案能够避免在GPU和外部设备之间来回拷贝数据，从而提高GPU处理数据的速度。

主权项

1.一种数据计算流图的GPU处理方法，其特征在于，包括：接收输入数据，并将所述输入数据存储至为计算流图中的第一源节点分配的第一显存块；获取所述第一源节点所指向的第一后继节点的第一表达式，从所述第一显存块读取所述输入数据，以及根据为所述计算流图中的源节点分配的显存块为所述第一后继节点分配第二显存块；在已获取到所述第一表达式所有需要输入的数据的情况下，将读取的所述输入数据输入至所述第一表达式进行计算，得到所述第一后继节点的结果，并将所述第一后继节点的结果存储至所述第二显存块；从所述第二显存块读取所述第一后继节点的结果，并根据读取的所述第一后继节点的结果得到所述输入数据按所述计算流图处理的结果，并输出所述输入数据的处理结果。