提交 0a4fdbcb authored 作者: Matthew Rocklin's avatar Matthew Rocklin

add do_constant_folding->False onto MPIRecv

上级 e30bdd89
...@@ -133,6 +133,9 @@ class MPIRecv(Op): ...@@ -133,6 +133,9 @@ class MPIRecv(Op):
def infer_shape(self, node, shapes): def infer_shape(self, node, shapes):
return [None, self.shape] return [None, self.shape]
def do_constant_folding(self, node):
return False
class MPIRecvWait(Op): class MPIRecvWait(Op):
""" """
An operation to wait on a previously received array using MPI An operation to wait on a previously received array using MPI
......
...@@ -16,3 +16,8 @@ def test_send(): ...@@ -16,3 +16,8 @@ def test_send():
sendnode = y.owner.inputs[0].owner sendnode = y.owner.inputs[0].owner
assert sendnode.op.dest == 1 assert sendnode.op.dest == 1
assert sendnode.op.tag == 11 assert sendnode.op.tag == 11
def test_can_make_function():
x = recv((5,5), 'float32', 0, 11)
y = x+1
assert theano.function([], [y])
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论