提交 d7d5401c authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Skip test for unimplemented case

上级 4de5d0c5
...@@ -31,6 +31,12 @@ class TestGpuCumsum(theano.tensor.tests.test_extra_ops.TestCumsumOp): ...@@ -31,6 +31,12 @@ class TestGpuCumsum(theano.tensor.tests.test_extra_ops.TestCumsumOp):
self.op_class = GpuCumsum self.op_class = GpuCumsum
def test_infer_shape(self): def test_infer_shape(self):
# GpuCumSum is only defined for float32 for now, so we skip it
# in the unsupported cases
gpucumsum_supported_dtypes = ('float32',)
if theano.config.floatX not in gpucumsum_supported_dtypes:
raise SkipTest('GpuCumSum not implemented for dtype %s'
% theano.config.floatX)
x = T.tensor3('x') x = T.tensor3('x')
a = np.random.random((3, 5, 2)).astype(theano.config.floatX) a = np.random.random((3, 5, 2)).astype(theano.config.floatX)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论