提交 66cd79b4 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add missing client check.

上级 426795d9
......@@ -41,6 +41,7 @@ def find_node(v, cls, ignore_clients=False):
return v.owner
elif (isinstance(v.owner.op, GpuFromHost) and
v.owner.inputs[0].owner is not None and
(ignore_clients or len(v.owner.inputs[0].clients) == 1) and
isinstance(v.owner.inputs[0].owner.op, HostFromGpu)):
return find_node(v.owner.inputs[0].owner.inputs[0], cls)
else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论