算子切分方法、电子设备、存储介质和计算机程序产品

本发明提供一种算子切分方法、电子设备、存储介质和计算机程序产品，所述方法包括：基于目标算子输入张量的切分轴，确定输出张量的切分轴；基于输入张量的批次轴尺寸以及输出张量的批次轴尺寸，确定是否校正输入张量的切分轴以及输出张量的切分轴；若是，则校正输入张量的切分轴以及输出张量的切分轴，并基于校正后的输入张量的切分轴以及校正后的输出张量的切分轴对目标算子进行算子切分；若否，则基于输入张量的切分轴以及输出张量的切分轴对目标算子进行算子切分。本发明能够实现在任意复杂场景下动态确定切分策略，无需为不同场景编写多个专用kernel，提高了算法的自适应性和通用性，实现了任意shape下MMA算子的自动化切分。