多GPU通信的方法及装置、存储介质、电子装置制造方法及图纸

本申请实施例中提供了多GPU通信的方法及装置、存储介质、电子装置，其中所述方法用于在一个主GPU和多个从GPU之间进行通信，所述主GPU与每个所述从GPU之间通过高速互联总线进行一对一通信，包括接收图像处理任务；将所述图像处理任务分割，得到多个图像处理子任务；根据所述从GPU的个数分配所述多个图像处理子任务；接收所述从GPU对所述图像处理任务的处理结果；将所述处理结果合并后拼接得到目标图像。采用本申请中的方案，实现了GPU协同完成复杂的绘图以及计算任务。杂的绘图以及计算任务。杂的绘图以及计算任务。

【技术实现步骤摘要】
多GPU通信的方法及装置、存储介质、电子装置


[0001]本申请涉及计算机图像处理技术，具体地，涉及多GPU通信的方法及装置、存储介质、电子装置。

技术介绍

[0002]在图像处理领域中，通常采用多片GPU之间相互协作，各自完成一部分图像处理的任务。
[0003]随着图像处理任务越来越复杂，GPU的整体性能成下降趋势。
[0004]针对相关技术中，当处理复杂图像处理任务时GPU整体性能不高的问题，目前尚未存在有效的解决方案。

技术实现思路

[0005]本申请实施例中提供了多GPU通信的方法及装置、存储介质、电子装置，以至少解决相关技术中当处理复杂图像处理任务时GPU整体性能不高的问题。
[0006]根据本申请实施例的第一个方面，提供了一种多GPU通信的方法，用于在一个主GPU和多个从GPU之间进行通信，所述主GPU与每个所述从GPU之间通过高速互联总线进行一对一通信，所述方法包括：接收图像处理任务；将所述图像处理任务分割，得到多个图像处理子任务；根据所述从GPU的个数分配所述多个图像处理子任务；接收所述从GPU对所述图像处理任务的处理结果；将所述处理结果合并后拼接得到目标图像。
[0007]根据本申请实施例的第二个方面，提供了一种多GPU通信的方法，用于在一个主GPU和多个从GPU之间进行通信，所述主GPU与每个所述从GPU之间通过高速互联总线进行一对一通信，所述方法包括：接收所述主GPU将图像处理任务分割后得到的多个图像处理子任务；按照预设方式处理所述多个图像处理子任务；将所述图像处理任务的处理结果发送至所述主GPU。
[0008]根据本申请实施例的第三个方面，提供了一种多GPU通信的装置，用于在一个主GPU和多个从GPU之间进行通信，所述主GPU与每个所述从GPU之间通过高速互联总线进行一对一通信，所述装置包括：第一接收模块，用于接收图像处理任务；分割模块，用于将所述图像处理任务分割，得到多个图像处理子任务；分配模块，用于根据所述从GPU的个数分配所述多个图像处理子任务；所述第一接收模块，还用于接收所述从GPU对所述图像处理任务的处理结果；拼接模块，用于将所述处理结果合并后拼接得到目标图像。
[0009]根据本申请实施例的第四个方面，提供了一种多GPU通信的装置，用于在一个主GPU和多个从GPU之间进行通信，所述主GPU与每个所述从GPU之间通过高速互联总线进行一对一通信，所述装置包括：第二接收模块，用于接收所述主GPU将图像处理任务分割后得到的多个图像处理子任务；处理模块，用于按照预设方式处理所述多个图像处理子任务；发送模块，用于将所述图像处理任务的处理结果发送至所述主GPU。
[0010]根据本申请实施例的第五个方面，提供了一种存储介质，所述存储介质中存储有
计算机程序，其中，所述计算机程序被设置为运行时执行上述任一项方法实施例中的步骤。
[0011]根据本申请实施例的第六个方面，提供了一种电子装置，包括存储器和处理器，所述存储器中存储有计算机程序，所述处理器被设置为运行所述计算机程序以执行上述任一项方法实施例中的步骤。
[0012]采用本申请实施例中提供的多GPU通信的方法及装置、存储介质、电子装置，通过在主GPU接收上层系统发送的绘图任务，完成自动的绘图任务切割，将部分绘图任务分配到多个从GPU，并同时管理多个从GPU的负载，当从GPU绘制完成后，接收绘图结果合并拼接成整幅图像输出。通过本申请达到GPU协同完成复杂绘图和计算任务的目的，实现了提高GPU整体性能的技术效果。
附图说明
[0013]此处所说明的附图用来提供对本申请的进一步理解，构成本申请的一部分，本申请的示意性实施例及其说明用于解释本申请，并不构成对本申请的不当限定。在附图中：
[0014]图1为本申请实施例中的系统结构示意图；
[0015]图2为本申请实施例中的多GPU通信的方法流程示意图；
[0016]图3为本申请实施例中的多GPU通信的装置结构示意图；
[0017]图4为本申请实施例中的多GPU通信的方法流程示意图；
[0018]图5为本申请实施例中的多GPU通信的装置结构示意图；
[0019]图6为本申请实施例中图像处理任务按绘图区均匀分配的示意图；
[0020]图7为本申请实施例中图像处理任务按绘图区非均匀分配的示意图。
具体实施方式
[0021]在实现本申请的过程中，专利技术人发现，在图像处理中通过多个GPU绘制图形然后进行显示，然而多个GPU无法并行协同处理任务，导致GPU整体性能不高。
[0022]针对上述问题，本申请实施例中提供了一种多GPU通信的方法，用于在一个主GPU和多个从GPU之间进行通信，所述主GPU与每个所述从GPU之间通过高速互联总线进行一对一通信，所述方法包括：接收图像处理任务；将所述图像处理任务分割，得到多个图像处理子任务；根据所述从GPU的个数分配所述多个图像处理子任务；接收所述从GPU对所述图像处理任务的处理结果；将所述处理结果合并后拼接得到目标图像。在本申请中采用多片GPU互联协同工作的方案，达到协同完成复杂绘图和计算任务的目的。
[0023]为了使本申请实施例中的技术方案及优点更加清楚明白，以下结合附图对本申请的示例性实施例进行进一步详细的说明，显然，所描述的实施例仅是本申请的一部分实施例，而不是所有实施例的穷举。需要说明的是，在不冲突的情况下，本申请中的实施例及实施例中的特征可以相互组合。
[0024]实施例一
[0025]本申请实施例一所提供的方法实施例可以在多GPU的系统或者类似架构的系统中执行。以运行在多GPU的系统上为例，图1是本专利技术实施例的多GPU通信的方法所实施的GPU系统结构示意图。如图1所示是一个主GPU，四个从GPU的GPU系统结构。主GPU与从GPU之间通过高速互联总线进行一对一通信。在所述主GPU接收上层的PCIE总线发送的绘图任务，将其
通过高速互联总线传输到其余三个从GPU中，三个从GPU也存在高速互联总线互通，在主GPU和从GPU之间可以通过共享RAM以及共享Cache共享绘图过程中需要的数据。
[0026]在本实施例中提供了多GPU通信的方法，用于在一个主GPU和多个从GPU之间进行通信，所述主GPU与每个所述从GPU之间通过高速互联总线进行一对一通信，如图2所示，该流程包括如下步骤：
[0027]步骤S201，接收图像处理任务；
[0028]步骤S202，将所述图像处理任务分割，得到多个图像处理子任务；
[0029]步骤S203，根据所述从GPU的个数分配所述多个图像处理子任务；
[0030]步骤S204，接收所述从GPU对所述图像处理任务的处理结果；
[0031]步骤S205，将所述处理结果合并后拼接得到目标图像。
[0032]上述多GPU通信的方法，通过在主GPU接收上层系统发送的绘图任务，完成自动的绘图任务切割，将部分绘本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种多GPU通信的方法，其特征在于，用于在一个主GPU和多个从GPU之间进行通信，所述主GPU与每个所述从GPU之间通过高速互联总线进行一对一通信，所述方法包括：接收图像处理任务；将所述图像处理任务分割，得到多个图像处理子任务；根据所述从GPU的个数分配所述多个图像处理子任务；接收所述从GPU对所述图像处理任务的处理结果；将所述处理结果合并后拼接得到目标图像。2.根据权利要求1所述的方法，其特征在于，所述根据所述从GPU的个数分配所述多个图像处理子任务包括：将所述从GPU待分配的图像处理子任务中的绘图数据包送至所述从GPU；在所述从GPU将所述绘图数据包接收完毕的情况下，接收信息应答包。3.根据权利要求2所述的方法，其特征在于，所述根据所述从GPU的个数分配所述多个图像处理子任务，还包括：向所述从GPU发送图像处理任务命令；在从GPU根据所述图像处理任务命令和所述绘图数据包执行完成所述图像处理任务的情况下，接收所述图像处理任务的完成应答包；向所述从GPU发送图像处理任务结果允许接收请求；根据所述允许接收请求，接收所述图像处理任务的处理结果。4.根据权利要求1所述的方法，其特征在于，所述根据所述从GPU的个数分配所述多个图像处理子任务，还包括：根据所述目标图像的坐标将绘图区，按照所述从GPU和所述主GPU的数目划分得到目标区域；将属于不同绘图区的绘图数据包和图像处理任务命令，发送至对应的所述从GPU。5.根据权利要求1所述的方法，其特征在于，所述主GPU与每个所述从GPU之间通过高速互联总线进行一对一通信包括：在每个所述从GPU之间通过高速互联总线进行一对一通信；和/或，所述主GPU与每个所述从GPU之间通过共享RAM共享执行所述图像处理任务过程中所需的数据；和/或，所述主GPU与每个所述从GPU之间通过共享Cache共享执行所述图像处理任务过程中所需的数据。6.根据权利要求1所述的方法，其特征在于，所述将所述处理结果合并后拼接得到目标图像包括：将所述处理结果与所述主GPU自身的图像处理结果合并后拼接得到目标图像。7.一种多GPU通信的方法，其特征在于，用于在一个主GPU和多个从GPU之间进行通信，所述主GPU与每个所述从GPU之间通过高速互联总线...

【专利技术属性】
技术研发人员：龙斌，
申请(专利权)人：长沙景美集成电路设计有限公司，
类型：发明
国别省市：