[发明专利]在TVM中加速调优收敛的调度方法、存储介质和电子设备

权利要求书

1.一种在TVM中加速调优收敛的调度方法，其特征在于，包括以下步骤：

S1，统计调度任务的类型，确定同类型调度任务的数量；

S2，首轮遍历调优，依次对调度任务进行首轮调优，记录调度任务对应的耗时；

S3，按照耗时长短为调度任务附上调度权重，耗时越长调度权重越重；

S4，计算耗时、同类型调度任务的数量、调度权重三者的乘积为同类型调度任务的得分；

S5，以所有调度任务的得分之和作为目标函数值与设定的目标阈值作比较，若目标函数值大于目标阈值，则跳至S6进行调度权重反馈，否则结束调优；

S6，调度权重反馈，对得分最大的同类型调度任务进行再次调优，并判断调优前后的耗时，若调优后的耗时比调优前的耗时短，则加重其调度权重作为奖励，否则减轻其调度权重作为惩罚；跳回至S4。

2.根据权利要求1所述的在TVM中加速调优收敛的调度方法，其特征在于，所述S2中，首轮遍历调优的具体内容为：初始化首轮遍历索引；对所述首轮遍历索引累加1，并依次对所述首轮遍历索引对应的调度任务进行调优直至所有调度任务完成首轮遍历调优。

3.根据权利要求1所述的在TVM中加速调优收敛的调度方法，其特征在于，所述S6中，调度权重反馈的具体内容为：

S61，选取得分最大的同类型调度任务，将其索引值赋给任务调度索引，将得分最大值赋给中间判断值；

S62，对所述得分最大的同类型调度任务进行再次调优，并记录调优后的耗时；

S63，将调优后的耗时与中间判断值进行比较，若调优后的耗时比中间判断值短，则加重对应的调度权重作为奖励，否则将减轻对应的调度权重作为惩罚。

4.根据权利要求3所述的在TVM中加速调优收敛的调度方法，其特征在于，所述奖励值大于惩罚值。

5.根据权利要求4所述的在TVM中加速调优收敛的调度方法，其特征在于，所述奖励内容为，在原调度权重的基础上加0.3；所述惩罚内容为，在原调度权重的基础上减0.2。

6.根据权利要求3所述的在TVM中加速调优收敛的调度方法，其特征在于，在所述S63中，还包括边界防护，当出现任一调度权重小于设定权重阈值时，进行调度任务进行重新洗牌。

7.根据权利要求6所述的在TVM中加速调优收敛的调度方法，其特征在于，所述重新洗牌的内容为：按照S3的方法，对当前的调度任务按照耗时长短为调度任务重新附上调度权重，耗时越长调度权重越重。

8.根据权利要求1所述的在TVM中加速调优收敛的调度方法，其特征在于，所述S1之前还包括：

S0，加载深度学习网络，并输入待检测图像，生成若干调度任务。

9.一种存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现权利要求1至7中任一项所述的调度方法。

10.一种电子设备，其特征在于，包括：

处理器；以及

存储器，用于存储所述处理器的可执行指令；

其中，所述处理器配置为经由执行所述可执行指令来执行权利要求1至7中任一项所述的调度方法。