提交 70c22839 authored 作者: Frederic Bastien's avatar Frederic Bastien

Disable check_py for cudnn tests.

上级 2ccd511d
...@@ -26,6 +26,11 @@ from .rnn_support import Model, GRU, LSTM, WrapperLayer ...@@ -26,6 +26,11 @@ from .rnn_support import Model, GRU, LSTM, WrapperLayer
from theano.configdefaults import SUPPORTED_DNN_CONV_ALGO_FWD from theano.configdefaults import SUPPORTED_DNN_CONV_ALGO_FWD
mode_with_gpu = mode_with_gpu.including()
mode_without_gpu = mode_without_gpu.including()
mode_with_gpu.check_py_code = False
mode_without_gpu.check_py_code = False
# If using float16, set CUDNN precision to float32 # If using float16, set CUDNN precision to float32
def set_precision(floatX): def set_precision(floatX):
......
...@@ -38,6 +38,8 @@ if theano.config.mode == 'FAST_COMPILE': ...@@ -38,6 +38,8 @@ if theano.config.mode == 'FAST_COMPILE':
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().excluding('gpu')
mode_with_gpu.check_py_code = False
mode_without_gpu.check_py_code = False
def test_dnn_conv_desc_merge(): def test_dnn_conv_desc_merge():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论