一种内核代码的循环展开方法、装置、设备及存储介质

本发明公开了一种内核代码的循环展开方法、装置、设备及存储介质，涉及深度学习技术领域，包括：在神经网络模型推理过程中，分别捕获神经网络模型各层输出张量的大小；根据各层输出张量的大小确定各层分别对应的当前展开因子；根据各当前展开因子生成当前统一计算设备架构代码，并对当前统一计算设备架构代码进行编译，得到编译后统一计算设备架构代码；利用编译后统一计算设备架构代码对内核代码进行循环展开，并获取当前循环展开时间；根据当前循环展开时间对各当前展开因子进行调整，得到各层分别对应的目标展开因子；根据各目标展开因子对内核代码进行循环展开。本发明降低了开发难度和成本，提高了展开因子与硬件架构的兼容性。