提交 ffa7a7ba authored 作者: affanv14's avatar affanv14 提交者: Mohammed Affan

skip tests if conditions are met

上级 db26d266
...@@ -1717,6 +1717,8 @@ class Grouped_conv_noOptim(unittest.TestCase): ...@@ -1717,6 +1717,8 @@ class Grouped_conv_noOptim(unittest.TestCase):
self.top_shape = [(5, 6, 3, 3), (4, 6, 3, 3), (3, 4, 3, 1), (2, 4, 5, 3)] self.top_shape = [(5, 6, 3, 3), (4, 6, 3, 3), (3, 4, 3, 1), (2, 4, 5, 3)]
self.filter_dilation = (1, 1) self.filter_dilation = (1, 1)
self.ref_mode = 'FAST_RUN' self.ref_mode = 'FAST_RUN'
if theano.config.cxx == "" or not theano.config.blas.ldflags:
raise SkipTest("CorrMM needs cxx and blas")
def test_fwd(self): def test_fwd(self):
for imshp, kshp, groups in zip(self.img_shape, self.kern_shape, self.num_groups): for imshp, kshp, groups in zip(self.img_shape, self.kern_shape, self.num_groups):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论