提交 013cf2af authored 作者: sebastien-j's avatar sebastien-j

Change mode for GPU_nstreams_limit

I am not sure about this one.
上级 d0e01971
......@@ -322,7 +322,8 @@ def test_GPU_nstreams_limit():
if theano.config.mode == "FAST_COMPILE":
mode = "FAST_RUN"
else:
mode = None
mode = theano.compile.get_default_mode()
mode.check_py = False
out = R.uniform(size=size, nstreams=nstreams, dtype='float32')
f = theano.function([], out, mode=mode)
return f()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论