一种任务调度方法、装置、设备及存储介质制造方法及图纸

本申请公开了一种任务调度方法、装置、设备及存储介质，其中，所述方法包括：获取第一任务申请显存的显存申请指令，其中，所述显存申请指令中包括待申请的显存容量；基于所述待申请的显存容量，确定待暂止运行的第二任务，其中，所述第一任务的显存使用优先级高于第二任务，所述第二任务使用的显存容量为可释放显存；在暂止运行所述第二任务的情况下，将所述第二任务的可释放显存分配给所述第一任务；在确定存在与所述第二任务匹配的空闲显存容量的情况下，继续运行所述第二任务。继续运行所述第二任务。继续运行所述第二任务。

【技术实现步骤摘要】
一种任务调度方法、装置、设备及存储介质


[0001]本申请实施例涉及计算机
，涉及但不限于一种任务调度方法、装置、设备及存储介质。

技术介绍

[0002]目前方案在显卡独占的调度平台中，可抢占式任务是占用整个显卡进行数据处理，被其他任务抢占后，可抢占式任务必须关闭重新进入调度队列。在显卡独占的调度平台中，由于程序的差异，有些任务虽然独占一块显卡，但是并不能利用显卡的全部显存和计算能力，造成了资源的浪费。显卡独占的可抢占式任务频繁关闭，造成调度、启动频繁，进而导致可抢占式任务的运行效率差。

技术实现思路

[0003]有鉴于此，本申请实施例提供一种任务调度方法、装置、设备及存储介质。
[0004]本申请实施例的技术方案是这样实现的：
[0005]第一方面，本申请实施例提供一种任务调度方法，所述方法包括：
[0006]获取第一任务申请显存的显存申请指令，其中，所述显存申请指令中包括待申请的显存容量；
[0007]基于所述待申请的显存容量，确定待暂止运行的第二任务，其中，所述第一任务的运行优先级高于本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种任务调度方法，所述方法包括：获取第一任务申请显存的显存申请指令，其中，所述显存申请指令中包括待申请的显存容量；基于所述待申请的显存容量，确定待暂止运行的第二任务，其中，所述第一任务的运行优先级高于第二任务，所述第二任务占用的显存为可释放显存；在暂止运行所述第二任务的情况下，将所述第二任务占用的可释放显存分配给所述第一任务；在确定存在与所述第二任务匹配的空闲显存容量的情况下，继续运行所述第二任务。2.如权利要求1所述的方法，所述在暂止运行所述第二任务的情况下，将所述第二任务占用的可释放显存分配给所述第一任务，包括：暂止运行所述第二任务；将所述第二任务占用的可释放显存中的显存数据迁移至指定内存空间；释放所述第二任务的可释放显存；将所述第二任务的可释放显存分配给所述第一任务。3.如权利要求2所述的方法，所述在确定存在与所述第二任务匹配的空闲显存容量的情况下，继续运行所述第二任务，包括：将所述第二任务存放至等待运行队列；确定存在与所述第二任务匹配的空闲显存；将所述第二任务对应的显存数据，从所述指定内存迁移至所述空闲显存；继续运行所述第二任务。4.如权利要求1所述的方法，与所述第二任务匹配的空闲显存属于第三任务所占用显卡的显存空间；其中，所述第三任务的运行优先级高于所述第二任务；所述在确定存在与所述第二任务匹配的空闲显存容量的情况下，继续运行所述第二任务，包括：基于所述第三任务所占用的显卡，继续运行所述第二任务。5.如权利要求3所述的方法，在所述在确定存在与所述第二任务匹配的空闲显存容量的情况下之前，所述方法还包括：定时扫描显存空间，获取所述显存空间的占用情况；基于所述显存空间的占用情况确定是否存在与所述第二任务匹配的空闲显存。6.如权利要求5所述的方法，所述方法应用于计算节点，所述...

【专利技术属性】
技术研发人员：陈旭，
申请(专利权)人：联想北京有限公司，
类型：发明
国别省市：