一种MPI消息通信算法的优化方法、装置、设备及可读介质

摘要

本发明公开了一种MPI消息通信算法的优化方法，包括以下步骤：获取并行应用程序调用的若干MPI函数，并将MPI函数使用的若干通信算法进行编号；基于若干通信算法的编号生成状态向量，将状态向量的变化作为动作集以构建动作价值函数；将应用程序单步运行速度作为奖励，通过强化学习算法基于奖励对动作价值函数进行迭代计算，以得到收敛后的最优结果；以及根据最优结果映射对应的通信算法作为最优通信算法，并基于最优通信算法调用底层基本通信模块进行交互。本发明还公开了对应的优化装置、计算机设备和可读存储介质。本发明将MPI函数通信算法的优化转换为强化学习的训练，实现为不同MPI函数选取最优的通信算法，缩短消息传输时间，提高并行计算效率。