提交 a909cbac authored 作者: Frederic's avatar Frederic

fix test crash with floatX=float64 and FAST_COMPILE

上级 ec69c0d9
...@@ -46,8 +46,8 @@ def pool_2d_i2n(input, ds=(2, 2), strides=None, pool_function=T.max, mode='ignor ...@@ -46,8 +46,8 @@ def pool_2d_i2n(input, ds=(2, 2), strides=None, pool_function=T.max, mode='ignor
def test_pooling(): def test_pooling():
if not cuda.dnn.dnn_available(): if not cuda.dnn.dnn_available():
raise SkipTest(cuda.dnn.dnn_available.msg) raise SkipTest(cuda.dnn.dnn_available.msg)
x = T.tensor4() x = T.ftensor4()
for func in (T.max, T.mean): for func in (T.max, T.mean):
for ws in (4, 5): for ws in (4, 5):
...@@ -57,8 +57,8 @@ def test_pooling(): ...@@ -57,8 +57,8 @@ def test_pooling():
out2 = pool_2d_i2n(x, ds=(ws, ws), strides=(stride, stride), out2 = pool_2d_i2n(x, ds=(ws, ws), strides=(stride, stride),
pool_function=func) pool_function=func)
f1 = theano.function([x], out1) f1 = theano.function([x], out1, mode=mode_with_gpu)
f2 = theano.function([x], out2) f2 = theano.function([x], out2, mode=mode_with_gpu)
data = numpy.random.normal(0, 1, (1, 10, 100, 100)).astype("float32") data = numpy.random.normal(0, 1, (1, 10, 100, 100)).astype("float32")
a = f1(data).__array__() a = f1(data).__array__()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论