高性能稀疏张量程序的自动生成方法及装置

本发明公开了一种高性能稀疏张量程序的自动生成方法及装置，该方法包括：基于领域定制语言将稀疏计算操作表示为稀疏计算描述，并进一步将稀疏计算描述转化为计算图；基于计算图分析得到计算特征并基于计算特征构建第一设计空间，从第一设计空间中的所有设计点中确定出目标点以基于目标点确定出目标原语配置；基于稀疏张量编译器以及目标原语配置，确定稀疏计算算子以生成高性能稀疏张量程序。可见，本发明能够将稀疏计算操作转变为计算图并构建第一设计空间和第二设计空间，通过搜索第二设计空间得到目标原语配置，有利于提高稀疏计算算子的确定准确性及效率，进而提高高性能稀疏张量程序的生成准确性和效率。