一种提高GEMM计算性能的方法及装置

摘要

本申请提供了一种提高GEMM计算性能的方法及装置，获取待优化的通用矩阵‑矩阵乘GEMM计算的参数，从至少一个历史的GEMM计算的参数中查询目标参数，目标参数为与待优化的GEMM计算的参数满足预设关系的参数。依据参数与优化方法的预设对应关系，确定所述目标参数对应的优化方法。并使用所述目标参数对应的优化方法，优化所述待优化的GEMM计算。其中，GEMM计算的参数基于参与GEMM计算的矩阵的大小确定。因为将参与待优化的GEMM计算的矩阵的特征作为优化待优化的GEMM计算的依据，所以，在使用深度卷积神经网络对目标进行检测的过程中，即使矩阵的规模小或者形状不规则，也能提高GEMM计算的性能。