提交 585b80bb authored 作者: Frederic's avatar Frederic

Skip cudnn test when the GPU is too old.

上级 667d2398
...@@ -577,8 +577,9 @@ def test_gemm_valid(): ...@@ -577,8 +577,9 @@ def test_gemm_valid():
yield t yield t
def test_dnn_valid(): def test_dnn_valid():
if cuda.device_properties(cuda.active_device_number())['major'] < 3:
raise SkipTest('Current GPU too old')
for t in _test_valid(GpuDnnConv, mode=theano_mode.including("cudnn")): for t in _test_valid(GpuDnnConv, mode=theano_mode.including("cudnn")):
yield t yield t
...@@ -658,6 +659,8 @@ def test_gemm_full(): ...@@ -658,6 +659,8 @@ def test_gemm_full():
def test_dnn_full(): def test_dnn_full():
if cuda.device_properties(cuda.active_device_number())['major'] < 3:
raise SkipTest('Current GPU too old')
for t in _test_full(GpuDnnConv, mode=theano_mode.including("cudnn")): for t in _test_full(GpuDnnConv, mode=theano_mode.including("cudnn")):
yield t yield t
...@@ -708,6 +711,8 @@ def test_gemm_subsample(): ...@@ -708,6 +711,8 @@ def test_gemm_subsample():
def test_dnn_subsample(): def test_dnn_subsample():
if cuda.device_properties(cuda.active_device_number())['major'] < 3:
raise SkipTest('Current GPU too old')
for t in _test_subsample(GpuDnnConv, theano_mode.including('cudnn')): for t in _test_subsample(GpuDnnConv, theano_mode.including('cudnn')):
yield t yield t
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论