[发明专利]一种基于任务窃取的任务调度方法及系统

摘要

本发明公开了一种基于任务窃取的任务调度方法及系统，该方法的实现包括：构造任务依赖图，将依赖节点作为回调函数注册至被依赖节点的回调容器中；为线程池中各线程分配一个无锁双端队列并置空，将根节点按照轮询方式放入各线程的无锁双端队列底部；若线程的无锁双端队列不为空，则从无锁双端队列底部取出节点并执行；若线程的无锁双端队列为空，则从其他线程的无锁双端队列顶部窃取节点，并将窃取的节点压入该线程的无锁双端队列底部，取出窃取的节点进行执行；在所有节点任务执行完成后，将任务依赖图中各节点的入度恢复到原始值，并结束对主线程的阻塞。本发明针对大型任务级并行应用程序，可以有效提高传统任务级并行应用程序的性能。

主权项

1.一种基于任务窃取的任务调度方法，其特征在于，包括：(1)将整体计算任务描述为由子任务节点与子任务节点间依赖边组成的任务依赖图，将依赖节点作为回调函数注册至被依赖节点的回调容器中；(2)获取所述任务依赖图中的根节点与叶子节点，为所有叶子节点添加一个虚拟依赖汇节点，所述虚拟依赖汇节点用于阻塞主线程；(3)为线程池中各线程分配一个无锁双端队列并置空，将所有根节点按照轮询方式放入各线程的无锁双端队列底部；(4)对于每个线程，若线程的无锁双端队列不为空，则从线程的无锁双端队列底部取出节点并执行节点中包含的任务，在任务执行结束后，执行节点中回调容器中的所有回调；若线程的无锁双端队列为空，则该线程尝试从其他线程的无锁双端队列顶部窃取节点，若窃取成功则将窃取的节点压入该线程的无锁双端队列底部，执行窃取节点中的任务以及窃取节点中回调容器中的所有回调；(5)在任务依赖图中所有节点中的任务均执行完成后，将任务依赖图中的各节点的入度恢复到原始值，并结束对主线程的阻塞。