[发明专利]基于MPI负载平衡方法、装置、计算机设备和存储介质

权利要求书

1.一种基于MPI负载平衡方法，其特征在于，包括：

获取第一时间，所述第一时间为n个节点从存储设备读取指定大小的地震数据所需的总时间，其中n1；

获取第二时间，所述第二时间为从节点0或所述存储设备到节点n，传输所述地震数据所需的时间；

获取第三时间，所述第三时间为从节点m到节点0，传输所述地震数据所需的时间，其中，m为n个节点中最先接收完所述地震数据，并开始计算的节点，且nm1；

获取第四时间，所述第四时间为计算单元完成一次所述地震数据处理及计算所需的时间；

将所述第三时间与所述第四时间进行求和处理，得到第五时间；

判断所述第一时间是否小于或等于所述第五时间；

当所述第一时间小于或等于所述第五时间时，判断所述第五时间是否大于所述第二时间；

当所述第五时间和大于所述第二时间时，在所述第一时间小于或等于所述第五时间的前提下，通过调节计算节点或者每一次传输地震数据的字节数，以使所述第五时间与所述第二时间的差值达到最小值，以实现负载平衡。

2.根据权利要求1所述的基于MPI负载平衡方法，其特征在于，在所述判断所述第一时间是否小于或等于所述第五时间的步骤之后；还包括：

当所述第一时间大于所述第五时间时，通过给定不同的MPI进程数并进行测试，计算出所述地震数据节点的最佳缓冲区大小；

根据所述地震数据节点的最佳缓冲区大小，对所述地震数据节点的缓冲区大小进行调节。

3.根据权利要求2所述的基于MPI负载平衡方法，其特征在于，在所述根据所述地震数据节点的最佳缓冲区大小，对所述地震数据节点的缓冲区大小进行调节的步骤之后，还包括：

对小粒度并行效率进行调整，以实现负载平衡。

4.根据权利要求1所述的基于MPI负载平衡方法，其特征在于，在所述判断所述第五时间是否大于所述第二时间的步骤之后，还包括：

当所述第五时间小于所述第二时间时，通过给定不同的MPI进程数并进行测试，计算出所述计算单元的最佳缓冲区大小；

根据所述计算单元的最佳缓冲区大小，对所述计算节点的缓冲区大小进行调节；

对小粒度并行效率进行调整，以实现负载平衡。

5.根据权利要求1所述的基于MPI负载平衡方法，其特征在于，所述获取第一时间，所述第一时间为n个节点从存储设备读取指定大小的地震数据所需的总时间，其中n1，包括：

对所述地震数据进行多次读取操作，得到地震数据读取速度；

根据地震数据读取分析算法，对所述地震数据读取速度进行计算，得到节点从所述存储设备读取数据的平均时间；

根据节点数n及所述平均时间，得到所述第一时间。

6.根据权利要求5所述的基于MPI负载平衡方法，其特征在于，所述对所述地震数据进行多次读取操作，得到地震数据读取速度的步骤，包括：

对segy格式的地震数据进行多次读取操作，得到所述地震数据读取速度。

7.根据权利要求1所述的基于MPI负载平衡方法，其特征在于，所述获取第二时间，所述第二时间为从节点0或所述存储设备到节点n，传输所述地震数据所需的时间，包括：

获取多次从节点0或所述存储设备到节点n，传输所述地震数据所需的时间，并对传输所述地震数据所需的时间进行加权平均，得到所述第二时间。