提交 3dd1bf0a authored 作者: Frederic's avatar Frederic

Fix GPU tests

上级 f706c95a
...@@ -14,11 +14,13 @@ if cuda.cuda_available == False: ...@@ -14,11 +14,13 @@ if cuda.cuda_available == False:
if theano.config.mode == 'FAST_COMPILE': if theano.config.mode == 'FAST_COMPILE':
mode_with_gpu = theano.compile.mode.get_mode('FAST_RUN').including('gpu') mode_with_gpu = theano.compile.mode.get_mode('FAST_RUN').including('gpu')
mode_without_gpu = theano.compile.mode.get_mode( # We should not exclude the 'gpu' tag, as some CPU opt are tagged
'FAST_RUN').excluding('gpu') # as GPU to make them run in fast_compile with gpu.
mode_without_gpu = theano.compile.mode.get_mode('FAST_RUN')
else: else:
mode_with_gpu = theano.compile.mode.get_default_mode().including('gpu') mode_with_gpu = theano.compile.mode.get_default_mode().including('gpu')
mode_without_gpu = theano.compile.mode.get_default_mode().excluding('gpu') mode_without_gpu = theano.compile.mode.get_default_mode()
def test_GpuCrossentropySoftmaxArgmax1HotWithBias(): def test_GpuCrossentropySoftmaxArgmax1HotWithBias():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论