一种面向矩阵加速单元的矩阵乘法优化方法及其系统

本申请公开了一种面向矩阵加速单元的矩阵乘法优化方法及其系统，方法包括：通过矩阵加速单元提供的矩阵乘指令集，采用线程重排、地址映射策略及多级流水线机制，共同构建微内核生成框架，生成候选微内核集合；针对输入矩阵进行特征提取，根据输入矩阵特征构建一个动态分块与调度策略构成的搜索空间，动态选择最优微内核并动态调整调度策略；构建性能预测模型，对候选微内核集合进行建模与剪枝，在搜索空间中优先选出最优调度策略组合，实现面向矩阵加速单元的矩阵乘法的优化。本发明方法显著提升了矩阵乘法优化的自动化程度和适应性，是一个可应用于多种矩阵乘架构并支持动态输入场景的通用高性能优化方案。