矩阵分片方法、装置、电子设备、存储介质及程序产品

本申请提供了一种矩阵分片方法、装置、电子设备、存储介质及程序产品；该方法包括：构建与待切分的第一矩阵对应的图模型，图模型中的节点表示第一矩阵对应的分片参数，图模型的每条边表示每条边连接的两个节点所表示的分片参数之间的参数调整关系；基于搜索算法和硬件参数，从图模型中确定从起始节点到预估结束节点的目标路径，其中，目标路径为从起始节点到预估结束节点的至少一条路径中总代价值最小的路径；根据目标路径，确定第一矩阵对应的分片方案。如此，在第一矩阵维度、硬件平台以及芯片发生变化时，通过搜索算法和硬件参数适应上述变化，使得上述方法对不同的矩阵维度、硬件平台以及芯片具有自适应能力，得到的分片方案更加灵活。