提交 e0dbbdbe authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Disable compute_test_values for mpi_roundtrip.

上级 8aa7567a
......@@ -4,6 +4,7 @@ from __future__ import absolute_import, print_function, division
from mpi4py import MPI
import theano
from theano.configparser import change_flags
from theano.tensor.io import send, recv, mpi_cmps
from theano.gof.sched import sort_schedule_fn
import numpy as np
......@@ -27,7 +28,8 @@ scheduler = sort_schedule_fn(*mpi_cmps)
mode = theano.Mode(optimizer=None,
linker=theano.OpWiseCLinker(schedule=scheduler))
if rank == 0:
with change_flags(compute_test_value='off'):
if rank == 0:
x = theano.tensor.matrix('x', dtype=dtype)
y = x + 1
send_request = send(y, 1, 11)
......@@ -45,7 +47,7 @@ if rank == 0:
# The parent test will look for "True" in the output
stdout.write(str(same))
if rank == 1:
if rank == 1:
y = recv(shape, dtype, 0, 11)
z = y * 2
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论