提交 7a9d92de authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix tests in float32.

上级 68dc51de
...@@ -167,8 +167,7 @@ class TestComputeTestValue(unittest.TestCase): ...@@ -167,8 +167,7 @@ class TestComputeTestValue(unittest.TestCase):
def test_scan(self): def test_scan(self):
""" """
Do not run this test as the compute_test_value mechanism is known not to work with Scan. Test the compute_test_value mechanism Scan.
TODO: fix scan to work with compute_test_value
""" """
orig_compute_test_value = theano.config.compute_test_value orig_compute_test_value = theano.config.compute_test_value
try: try:
...@@ -177,7 +176,7 @@ class TestComputeTestValue(unittest.TestCase): ...@@ -177,7 +176,7 @@ class TestComputeTestValue(unittest.TestCase):
k = T.iscalar("k") k = T.iscalar("k")
A = T.vector("A") A = T.vector("A")
k.tag.test_value = 3 k.tag.test_value = 3
A.tag.test_value = numpy.random.rand(5) A.tag.test_value = numpy.random.rand(5).astype(config.floatX)
def fx(prior_result, A): def fx(prior_result, A):
return prior_result * A return prior_result * A
...@@ -204,7 +203,7 @@ class TestComputeTestValue(unittest.TestCase): ...@@ -204,7 +203,7 @@ class TestComputeTestValue(unittest.TestCase):
k = T.iscalar("k") k = T.iscalar("k")
A = T.matrix("A") A = T.matrix("A")
k.tag.test_value = 3 k.tag.test_value = 3
A.tag.test_value = numpy.random.rand(5,3) A.tag.test_value = numpy.random.rand(5,3).astype(config.floatX)
def fx(prior_result, A): def fx(prior_result, A):
return T.dot(prior_result, A) return T.dot(prior_result, A)
...@@ -240,7 +239,7 @@ class TestComputeTestValue(unittest.TestCase): ...@@ -240,7 +239,7 @@ class TestComputeTestValue(unittest.TestCase):
k = T.iscalar("k") k = T.iscalar("k")
A = T.matrix("A") A = T.matrix("A")
k.tag.test_value = 3 k.tag.test_value = 3
A.tag.test_value = numpy.random.rand(5,3) A.tag.test_value = numpy.random.rand(5,3).astype(config.floatX)
def fx(prior_result, A): def fx(prior_result, A):
return T.dot(prior_result, A) return T.dot(prior_result, A)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论