提交 2bdfda4a authored 作者: Frederic Bastien's avatar Frederic Bastien

Skip cudnn test if cudnn isn't there.

上级 ea1d341b
...@@ -2288,6 +2288,11 @@ class Cudnn_grouped_conv(Grouped_conv_noOptim): ...@@ -2288,6 +2288,11 @@ class Cudnn_grouped_conv(Grouped_conv_noOptim):
conv_gradw_op = dnn.GpuDnnConvGradW conv_gradw_op = dnn.GpuDnnConvGradW
conv_gradi_op = dnn.GpuDnnConvGradI conv_gradi_op = dnn.GpuDnnConvGradI
def __init__(self, *args, **kwargs):
if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg)
super(Cudnn_grouped_conv, self).__init__(*args, **kwargs)
class Cudnn_grouped_conv3d(Grouped_conv3d_noOptim): class Cudnn_grouped_conv3d(Grouped_conv3d_noOptim):
mode = mode_with_gpu.excluding('conv_gemm') mode = mode_with_gpu.excluding('conv_gemm')
...@@ -2295,6 +2300,11 @@ class Cudnn_grouped_conv3d(Grouped_conv3d_noOptim): ...@@ -2295,6 +2300,11 @@ class Cudnn_grouped_conv3d(Grouped_conv3d_noOptim):
conv_gradw_op = dnn.GpuDnnConvGradW conv_gradw_op = dnn.GpuDnnConvGradW
conv_gradi_op = dnn.GpuDnnConvGradI conv_gradi_op = dnn.GpuDnnConvGradI
def __init__(self, *args, **kwargs):
if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg)
super(Cudnn_grouped_conv3d, self).__init__(*args, **kwargs)
def test_dnn_spatialtf(): def test_dnn_spatialtf():
if not dnn.dnn_available(test_ctx_name): if not dnn.dnn_available(test_ctx_name):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论