[发明专利]图计算的处理方法、装置、电子设备及存储介质有效
申请号: | 202010074245.4 | 申请日: | 2020-01-22 |
公开(公告)号: | CN111292223B | 公开(公告)日: | 2023-07-25 |
发明(设计)人: | 汪洋;宋勋超;袁鹏程;王益飞;张海平 | 申请(专利权)人: | 北京百度网讯科技有限公司 |
主分类号: | G06T1/20 | 分类号: | G06T1/20 |
代理公司: | 北京品源专利代理有限公司 11332 | 代理人: | 孟金喆 |
地址: | 100085 北京市*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 计算 处理 方法 装置 电子设备 存储 介质 | ||
1.一种图计算的处理方法,其特征在于,包括:
在中央处理器CPU中配置执行引擎,所述执行引擎用于主导图计算的执行;
配置用于在图形处理器GPU中执行的接口函数,所述接口函数用于进行图计算中的参数处理或反馈;
在执行图计算时,所述执行引擎通过图计算接口调用接口函数;
所述图形处理器GPU通过多个并行线程执行所述接口函数,多个并行的接口函数用于并行的对多个图节点进行处理,所述多个并行线程将所述接口函数的执行结果反馈至所述执行引擎;
所述执行引擎根据所述执行结果完成图计算。
2.根据权利要求1所述的图计算的处理方法,其特征在于,所述执行引擎通过所述图计算接口调用接口函数,包括:
执行引擎通过调用收集方向前置判断接口函数gather_edge获取入边参数;
执行引擎根据所述入边参数进行目标节点预设参数的收敛,得到所述目标节点的更新数据;
执行引擎通过调用数据变更应用接口函数apply对所述目标节点的预设参数进行更新。
3.根据权利要求2所述的图计算的处理方法,其特征在于,在执行引擎通过调用数据变更应用接口函数对所述目标节点的预设参数进行更新之后,还包括:
执行引擎通过调用发散方向前置判断接口函数scatter_edge获取节点的出边参数;
执行引擎通过调用发散接口函数scatter,将所述更新数据更新到出边的对端节点。
4.根据权利要求2所述的图计算的处理方法,其特征在于,执行引擎根据所述入边参数进行目标节点预设参数的收敛,得到所述目标节点的更新数据,包括:
执行引擎通过访问数据库获取目标节点的入边临接索引;
执行引擎根据所述入边临接索引进行目标节点预设参数的收敛,得到所述目标节点的更新数据。
5.根据权利要求4所述的图计算的处理方法,其特征在于,执行引擎根据所述入边临接索引进行目标节点预设参数的收敛,得到所述目标节点的更新数据,包括:
执行引擎根据所述入边临接索引获取每个入边;
执行引擎通过调用集合接口函数gather获取所述每个入边的对端节点的预设参数;
执行引擎调用图形处理器GPU中的累加接口函数sum对所述入边的对端节点的预设参数进行累加,得到所述目标节点的更新数据。
6.根据权利要求1所述的图计算的处理方法,其特征在于,所述执行引擎通过所述图计算接口调用接口函数,包括:
所述执行引擎根据图形处理器GPU的内存空间大小对待处理的图数据进行分块,得到多个节点块,每个节点块与图形处理器GPU的内存空间大小匹配;
所述执行引擎通过所述图计算接口调用接口函数,依次对所述多个节点块进行图计算。
7.一种图计算的处理装置,其特征在于,包括:
配置模块,用于在中央处理器CPU中配置执行引擎,所述执行引擎用于主导图计算的执行;配置用于在图形处理器GPU中执行的接口函数,所述接口函数用于进行图计算中的参数处理或反馈;
执行引擎模块,用于在执行图计算时,通过图计算接口调用接口函数;
接口调用模块,用于所述图形处理器GPU通过多个并行线程执行所述接口函数,多个并行的接口函数用于并行的对多个图节点进行处理,所述多个并行线程将所述接口函数的执行结果反馈至所述执行引擎;
所述执行引擎模块还用于,根据所述执行结果完成图计算。
8.根据权利要求7所述的图计算的处理装置,其特征在于,所述执行引擎模块用于:
执行引擎通过调用收集方向前置判断接口函数gather_edge获取入边参数;
执行引擎根据所述入边参数进行目标节点预设参数的收敛,得到所述目标节点的更新数据;
执行引擎通过调用数据变更应用接口函数apply对所述目标节点的预设参数进行更新。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京百度网讯科技有限公司,未经北京百度网讯科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202010074245.4/1.html,转载请声明来源钻瓜专利网。