(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202211275747.9 (22)申请日 2022.10.18 (71)申请人 杭州宇泛智能科技有限公司 地址 311121 浙江省杭州市余杭区仓前街 道绿汀路1号1号楼6 58室 (72)发明人 郭华朋　王亚楠　胡兵　张波　 (74)专利代理 机构 北京清辰科创知识产权代理 事务所(普通 合伙) 1613 3 专利代理师 彭一波 (51)Int.Cl. G06F 9/48(2006.01) G06F 9/54(2006.01) G06N 3/04(2006.01) G06N 3/08(2006.01) (54)发明名称 任务调度方法、 装置、 电子设备及存 储介质 (57)摘要 本申请提供一种任务调度方法、 装置、 电子 设备及计算机可读存储介质。 该任务调度方法包 括： 基于用户优 先级等级和系统时间计算训练任 务的权重值； 当所述训练任务对应资源组存在 Zset队列时， 将所述训练任务写入所述 资源组的 Zset队列中； 当所述训练任务对应资源组不存在 Zset队列 时， 创建所述资源组的Zset队列， 并将 所述训练任务写入所述 资源组的Zset队列中； 当 不同资源组的Zset队列存在排队任务时， 获取不 同资源组的Zset队列中权重值最大的训练任务； 将权重值最大的训练任务提交至调度服务器。 本 申请实现了调度任务自动排队， 解决了分布式任 务调度的高性能与数据可靠性之间的平衡问题， 实现了训练任务在队列中位置的调整。 权利要求书2页 说明书8页 附图2页 CN 115469988 A 2022.12.13 CN 115469988 A 1.一种任务调度方法， 其特 征在于， 所述方法包括： 基于用户优先级等级和系统时间计算训练任务的权 重值； 当所述训练任务对应资源组存在Zset队列时， 将所述训练任务写入所述资源组的Zset 队列中； 当所述训练任务对应资源组不存在 Zset队列时， 创建所述资源组的Zset队列， 并将 所述训练任务写入所述资源组的Zset队列中； 当不同资源组的Zset队列存在排队任务时， 获取不同资源组的Zset队列中权重值最大 的训练任务； 将权重值最大的训练任务 提交至调度服 务器。 2.根据权利要求1所述的任务调度方法， 其特征在于： 所述训练任务的权重值等于权重 因子*用户优先级等级+系统时间。 3.根据权利要求1所述的任务调度方法， 其特征在于： 在所述计算训练任务的权重值之 前， 所述方法还 包括： 创建训练任务； 判断所述训练任务的训练参数 是否有效； 当所述训练参数有效时， 查询并获取用户优先级等级和系统时间。 4.根据权利要求3所述的任务调度 方法， 其特征在于： 所述训练参数包括资源组、 GPU卡 数以及用户登录信息 。 5.根据权利要求1～4中任一项所述的任务调度方法， 其特征在于： 所述方法还包括调 整训练任务在对应资源组的Zset队列中的位置的步骤， 其具体实现过程 为： 通过修改权 重值调整训练任务在对应资源组的Zset队列中的位置 。 6.根据权利要求5所述的任务调度方法， 其特征在于： 调整训练任务在对应资源组的 Zset队列中的位置的具体操作为： 计算并更新被调整的训练任务的权 重值； 其中， 当调整至队列头部时， 被调整的训练任务的权重值等于队列头部训练任务的权 重值+1； 当调整至队列尾部时， 被调整的训练任务的权重值等于队列尾部训练任务的权重值－ 1； 当调整至队列任意中间位置时， 被调整的训练任务的权重值等于(目标位置前一训练 任务的权 重值+目标位置后一训练任务的权 重值)/2。 7.根据权利要求1所述的任务调度方法， 其特征在于： 在将权重值最大的训练任务提交 至调度服 务器之前， 所述方法还 包括： 判断所述调度服务器硬件资源是否满足需求， 在满足需求 时将权重值最大的训练任务 提交至调度服 务器。 8.一种任务调度装置， 其特 征在于， 所述装置包括： 计算单元， 被配置为基于用户优先级等级和系统时间计算训练任务的权 重值； 任务写入单元， 被配置为当所述训练任务对应资源组存在Zset队列 时， 将所述训练任 务写入所述资源组的Zset队列中； 当所述训练任务对应资源组不存在Zset队列时， 创建所 述资源组的Zset队列， 并将所述训练任务写入所述资源组的Zset队列中； 获取单元， 被配置为当不同资源组的Zset队列存在排队任务时， 获取不同资源组的权　利　要　求　书 1/2 页 2 CN 115469988 A 2Zset队列中权 重值最大的训练任务； 提交单元， 被配置为将权 重值最大的训练任务 提交至调度服 务器。 9.一种电子设备， 所述电子设备包括处理器和存储器， 所述存储器中存储有计算机程 序， 其特征在于： 所述处理器调用所述存储器中的计算机程序时执行权利要求1～7中任一 项所述任务调度方法中的步骤。 10.一种计算机可读存储介质， 其上存储有计算机程序， 其特征在于： 所述计算机程序 被处理器进行加载， 以执 行权利要求1～7中任一项所述任务调度方法中的步骤。权　利　要　求　书 2/2 页 3 CN 115469988 A 3