[发明专利]任务调度方法、装置和电子设备

权利要求书

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训练任务描述文件中包括MPC训练任务的任务名称和执行机构；以及将所述MPC训练任务描述文件拆分为子任务描述文件；其中，所述子任务描述文件的数量由所述执行机构的数量确定，所述子任务描述文件用于描述所述MPC训练任务拆分获得的子任务；

调度模块，用于根据所述子任务描述文件，对所述MPC训练任务拆分获得的子任务进行调度；

调度拦截模块，用于记录调度成功的子任务的信息；以及根据记录的信息，确定所述MPC训练任务拆分获得的子任务均调度成功之后，将所述拆分获得的子任务下发给对应的执行机构。

8.根据权利要求7所述的装置，其中，所述子任务描述文件中包括子任务的名称、子任务关联的MPC训练任务的名称和所述MPC训练任务拆分的子任务的数量；

所述调度拦截模块，具体用于记录调度成功的子任务的名称、调度成功的子任务关联的MPC训练任务的名称和所述MPC训练任务拆分的子任务的数量。