提交 294d7b6e authored 作者: Frederic Bastien's avatar Frederic Bastien

enable gpu convolution test as their crash have been fixed.

上级 ca383b23
...@@ -2,9 +2,6 @@ import sys, time ...@@ -2,9 +2,6 @@ import sys, time
import numpy import numpy
import theano import theano
from nose.plugins.skip import SkipTest
raise SkipTest('SKIP TO PREVENT THE BUILDBOT FROM CRASHING. THERE IS A DIFFICULT BUG TO FIX WITH MEMORY LEAK AND/OR WHEN Cuda_Ndarray alloc fail!')
# Skip test if cuda_ndarray is not available. # Skip test if cuda_ndarray is not available.
from nose.plugins.skip import SkipTest from nose.plugins.skip import SkipTest
import theano.sandbox.cuda as cuda_ndarray import theano.sandbox.cuda as cuda_ndarray
...@@ -110,6 +107,7 @@ def _params_allgood(ishape, kshape, mode, subsample=(1,1), img_stride=(1,1), ker ...@@ -110,6 +107,7 @@ def _params_allgood(ishape, kshape, mode, subsample=(1,1), img_stride=(1,1), ker
rval = False rval = False
if rval: if rval:
rval = numpy.allclose(cpuval, gpuval, rtol = rtol) rval = numpy.allclose(cpuval, gpuval, rtol = rtol)
assert numpy.all(numpy.isfinite(gpuval))
except NotImplementedError, e: except NotImplementedError, e:
print >> sys.stdout, '_params_allgood Failed allclose', e print >> sys.stdout, '_params_allgood Failed allclose', e
rval = False rval = False
...@@ -179,7 +177,7 @@ def exec_conv(version, shapes, verbose, random, mode, print_=None, rtol=1e-5, on ...@@ -179,7 +177,7 @@ def exec_conv(version, shapes, verbose, random, mode, print_=None, rtol=1e-5, on
nb_tests+=1 nb_tests+=1
if nb_failed>0: if nb_failed>0:
print "nb_failed",nb_failed,"on",nb_tests, "failed_version",failed_version, "failed_id",failed_id print "nb_failed",nb_failed,"on",nb_tests, "failed_version",failed_version, "failed_id",failed_id
assert nb_failed==0 assert nb_failed==0, nb_failed
else: else:
print 'Executed',nb_tests,'different shapes' print 'Executed',nb_tests,'different shapes'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论