在TVM中加速调优收敛的调度方法、存储介质和电子设备

基本信息

申请号 CN202111082553.2 申请日 -
公开(公告)号 CN113687936A 公开(公告)日 2021-11-23
申请公布号 CN113687936A 申请公布日 2021-11-23
分类号 G06F9/48 分类 计算;推算;计数;
发明人 姜枞聪;樊一超;曹航;李冠华 申请(专利权)人 杭州云栖智慧视通科技有限公司
代理机构 杭州信义达专利代理事务所(普通合伙) 代理人 陈继算
地址 310000 浙江省杭州市西湖区转塘科技经济区块16号2幢401室
法律状态 -

摘要

摘要 本发明公开了一种在TVM中加速调优收敛的调度方法,包括以下步骤:统计调度任务的类型,确定同类型调度任务的数量;首轮遍历调优,记录调度任务对应的耗时;按照耗时长短为调度任务附上调度权重;计算同类型调度任务的得分;以所有调度任务的得分之和作为目标函数值与设定的目标阈值作比较,若目标函数值大于目标阈值,则进行调度权重反馈;调度权重反馈,进行再次调优,并判断调优前后的耗时比较调整调度权重;本发明还同步公开一种、存储介质和电子设备。本发明对调优后的耗时与调优前进行比较,并进行相应的调度权重奖励与惩罚,使调度能够更加快速地收敛。