提交 2a339bb1 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Skip the CorrMM tests if there is no blas.

上级 540e7d38
...@@ -256,6 +256,11 @@ class BaseTestConv2d(unittest.TestCase): ...@@ -256,6 +256,11 @@ class BaseTestConv2d(unittest.TestCase):
class TestCorrConv2d(BaseTestConv2d): class TestCorrConv2d(BaseTestConv2d):
def setUp(self):
if theano.config.blas.ldflags == "":
raise SkipTest()
return super(TestCorrConv2d, self).setUp()
def tcase(self, i, f, s, b, flip, provide_shape): def tcase(self, i, f, s, b, flip, provide_shape):
o = self.get_output_shape(i, f, s, b) o = self.get_output_shape(i, f, s, b)
self.run_fwd(inputs_shape=i, filters_shape=f, subsample=s, self.run_fwd(inputs_shape=i, filters_shape=f, subsample=s,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论