如何结合延迟函数调用的顺序执行和并行执行?

时间:2019-02-07 13:22:05

标签: dask dask-distributed dask-delayed

我被困在一个陌生的地方。我有一堆要按特定顺序执行的延迟函数调用。虽然并行执行很简单:

res = client.compute([myfuncs])
res = client.gather(res)

我似乎找不到一种以非阻塞方式依次执行它们的方法。

这是一个最小的例子:

import numpy as np
from time import sleep
from datetime import datetime

from dask import delayed
from dask.distributed import LocalCluster, Client


@delayed
def dosomething(name):
    res = {"name": name, "beg": datetime.now()}
    sleep(np.random.randint(10))
    res.update(rand=np.random.rand())
    res.update(end=datetime.now())
    return res


seq1 = [dosomething(name) for name in ["foo", "bar", "baz"]]
par1 = dosomething("whaat")
par2 = dosomething("ahem")
pipeline = [seq1, par1, par2]

鉴于上面的示例,我想并行运行seq1par1par2,但是seq1的组成部分:“ foo”,“ bar ”和“ baz”。

1 个答案:

答案 0 :(得分:1)

您绝对可以欺骗并向您的函数添加可选的依赖项,如下所示:

@dask.delayed
def dosomething(name, *args):
     ...

这样,即使您在下次运行该函数时都不使用一个结果,也可以使任务彼此依赖:

inputs = ["foo", "bar", "baz"]
seq1 = [dosomething(inputs[0])]
for bit in inputs[1:]:
    seq1.append(dosomething(bit, seq1[-1]))

或者,您可以阅读分布式调度程序的“未来”界面,从而可以实时监视任务的进度。