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