提交 7286408b authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Don't fail when no theano flags in environment

上级 e0dbbdbe
......@@ -47,7 +47,7 @@ def test_mpi_roundtrip():
raise SkipTest('MPI not enabled')
theano_root = theano.__file__.split('__init__')[0]
env = os.environ.copy()
flags = env['THEANO_FLAGS']
flags = env.get('THEANO_FLAGS', '')
keep_flags = ','.join((f for f in flags.split(',') if not f.startswith('init_gpu_device')))
env['THEANO_FLAGS'] = keep_flags
p = subprocess.Popen("mpiexec -np 2 python " + theano_root +
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论