提交 1f59b11e authored 作者: Matthew Rocklin's avatar Matthew Rocklin

sendwait now returns a variable

I suspect that this is necessary to hold as an output
上级 97aa44e0
...@@ -224,11 +224,12 @@ class MPISendWait(Op): ...@@ -224,11 +224,12 @@ class MPISendWait(Op):
return hash(self.type) return hash(self.type)
def make_node(self): def make_node(self):
return gof.Apply(self, [theano.Generic()], []) return gof.Apply(self, [theano.Generic()], [theano.Generic()])
def perform(self, node, inp, out): def perform(self, node, inp, out):
request = inp[0][0] request = inp[0][0]
request.wait() request.wait()
out[0][0] = True
def __str__(self): def __str__(self):
return "MPISendWait" return "MPISendWait"
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论