任务调度方法、装置和电子设备.pdf

(19)国家知识产权局(12)发明专利申请(10)申请公布号CN114721804A(43)申请公布日2022.07.08(21)申请号202210397549.3(22)申请日2022.04.15(71)申请人支付宝（杭州）信息技术有限公司地址310007浙江省杭州市西湖区西溪路556号8层B段801-11(72)发明人陆宇飞曹辰郭石磊张本宇王磊(74)专利代理机构北京汇思诚业知识产权代理有限公司11444专利代理师焦志刚(51)Int.Cl.G06F9/48(2006.01)权利要求书3页说明书14页附图7页(54)发明名称任务调度方法、装置和电子设备(57)摘要本说明书实施例提出了一种任务调度方法、装置和电子设备，其中，上述任务调度方法中，获取MPC训练任务描述文件之后，将MPC训练任务描述文件拆分为子任务描述文件，然后根据上述子任务描述文件，对上述MPC训练任务拆分获得的子任务进行调度，记录调度成功的子任务的信息。最后，根据记录的信息，确定上述MPC训练任务拆分获得的子任务均调度成功之后，将拆分获得的子任务下发给对应的执行机构，从而可以实现将MPC训练任务拆分为多个K8s支持的单方任务来执行，达到运行MPC多方任务的效果，使K8s适用于MPC场景；并且上述方法可以实现将同一个MPC训练任务下的多个子任务一并下发，提高MPC训练任务的执行成功率，避免执行机构的算力浪费。CN114721804ACN114721804A权利要求书1/3页1.一种任务调度方法，包括：获取安全多方计算MPC训练任务描述文件；其中，所述MPC训练任务描述文件中包括MPC训练任务的任务名称和执行机构；将所述MPC训练任务描述文件拆分为子任务描述文件；其中，所述子任务描述文件的数量由所述执行机构的数量确定，所述子任务描述文件用于描述所述MPC训练任务拆分获得的子任务；根据所述子任务描述文件，对所述MPC训练任务拆分获得的子任务进行调度；记录调度成功的子任务的信息；根据记录的信息，确定所述MPC训练任务拆分获得的子任务均调度成功之后，将所述拆分获得的子任务下发给对应的执行机构。2.根据权利要求1所述的方法，其中，所述子任务描述文件中包括子任务的名称、子任务关联的MPC训练任务的名称和所述MPC训练任务拆分的子任务的数量；所述记录调度成功的子任务的信息包括：记录调度成功的子任务的名称、调度成功的子任务关联的MPC训练任务的名称和所述MPC训练任务拆分的子任务的数量。3.根据权利要求2所述的方法，其中，所述根据记录的信息，确定所述MPC训练任务拆分获得的子任务均调度成功包括：根据调度成功的子任务关联的MPC训练任务的名称，获取所述MPC训练任务拆分的子任务中调度成功的子任务的数量；如果所述拆分的子任务中调度成功的子任务的数量等于所述拆分的子任务的数量，则确定所述MPC训练任务拆分获得的子任务均调度成功。4.根据权利要求1所述的方法，其中，所述子任务描述文件中包括：子任务的执行机构；所述根据所述子任务描述文件，对所述MPC训练任务拆分获得的子任务进行调度包括：根据所述子任务描述文件中包括的执行机构，从所述执行机构包括的机器中选择适合执行所述子任务的机器。5.根据权利要求1‑4任意一项所述的方法，其中，所述将所述拆分获得的子任务下发给对应的执行机构之后，还包括：监听每个执行机构上子任务的运行状态；如果所有执行机构上的子任务均执行成功，则将所述MPC训练任务的状态标记为执行成功。6.根据权利要求5所述的方法，其中，所述监听每个执行机构上子任务的运行状态之后，还包括：如果至少一个执行机构上的子任务执行失败，则将所述MPC训练任务的状态标记为执行失败；向其他执行机构发送子任务中止指令，以中止所述其他执行机构上子任务的执行；其中，所述其他执行机构包括所述MPC训练任务的执行机构中除所述至少一个执行机构之外的执行机构。7.一种任务调度装置，包括：任务控制模块，用于获取安全多方计算MPC训练任务描述文件；其中，所述MPC训练任务2CN114721804A权利要求书2/3页描述文件中包括MPC训练任务的任务名称和执行机构；以及将所述MPC训练任务描述文件拆分为子任务描述文件；其中，所述子任务描述文件的数量由所述执行机构的数量确定，所述子任务描述文件用于描述所述MPC训练任务拆分获得的子任务；调度模块，用于根据所述子任务描述文件，对所述MPC训练任务拆分获得的子任务进行调度；调度拦截模块，用于记录调度成功的子任务的信息；以及根据记录的信息，确定所述MPC训练任务拆分获得的子任务均调度成功之后，将所述拆分获得的子任务下发给对应的执行机构。8.根据权利要求7所述的装置，其中，所述子任务描述文件中包括子任务的名称、子任务关联的MPC训练任务的名称和所述MPC训练任务拆分的子任务的数