提交 172e699c authored 作者: Gijs van Tulder's avatar Gijs van Tulder

gpuarray.mode_without_gpu should also exclude gpu.

上级 e722f71f
...@@ -17,7 +17,7 @@ test_ctx_name = None ...@@ -17,7 +17,7 @@ test_ctx_name = None
if theano.config.mode == 'FAST_COMPILE': if theano.config.mode == 'FAST_COMPILE':
mode_with_gpu = theano.compile.mode.get_mode('FAST_RUN').including('gpuarray').excluding('gpu') mode_with_gpu = theano.compile.mode.get_mode('FAST_RUN').including('gpuarray').excluding('gpu')
mode_without_gpu = theano.compile.mode.get_mode('FAST_RUN').excluding('gpuarray') mode_without_gpu = theano.compile.mode.get_mode('FAST_RUN').excluding('gpuarray', 'gpu')
else: else:
mode_with_gpu = theano.compile.mode.get_default_mode().including('gpuarray').excluding('gpu') mode_with_gpu = theano.compile.mode.get_default_mode().including('gpuarray').excluding('gpu')
mode_without_gpu = theano.compile.mode.get_default_mode().excluding('gpuarray') mode_without_gpu = theano.compile.mode.get_default_mode().excluding('gpuarray', 'gpu')
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论