提交 0177c84f authored 作者: Frederic's avatar Frederic

Skip dnn test if it isn't avail

上级 de6791fd
......@@ -860,7 +860,8 @@ def conv_grad(mode, bs, ch, nf, rImg1, rImg2, rFlt1, rFlt2, subsample, op):
def test_conv_grads():
if cuda.device_properties(cuda.active_device_number())['major'] < 3:
if (not cuda.dnn.dnn_available() or
cuda.device_properties(cuda.active_device_number())['major'] < 3):
ops = [gemm_op]
else:
ops = [gemm_op, dnn_op]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论