提交 a7b22194 authored 作者: Caglar's avatar Caglar

Fixed the typos in the test.

上级 75f8fee4
...@@ -5,6 +5,7 @@ import unittest ...@@ -5,6 +5,7 @@ import unittest
import numpy import numpy
# 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
from nose.tools import assert_raises
import theano import theano
from theano.compile.pfunc import pfunc from theano.compile.pfunc import pfunc
...@@ -92,33 +93,30 @@ def test_local_gpu_contiguous_gpu_contiguous(): ...@@ -92,33 +93,30 @@ def test_local_gpu_contiguous_gpu_contiguous():
def test_local_assert_no_cpu_op(): def test_local_assert_no_cpu_op():
#opt.assert_no_cpu_op = "raise"
numpy.random.seed(1) numpy.random.seed(1)
m = numpy.random.uniform(-1, 1, (10, 10)).astype("float32") m = numpy.random.uniform(-1, 1, (10, 10)).astype("float32")
ms = theano.shared(m, name="m_shared") ms = cuda.shared_constructor(m, name="m_shared")
z = ms**2 + 3 out = theano.tensor.tanh(ms**2 + 3)
def perform_assert_no_cpu_test(out, flag="ignore"): mode_local_assert = mode_with_gpu.including("local_assert_no_cpu_op")
mode_local_assert = mode_local_assert.excluding("local_gpu_elemwise_1")
mode_local_assert = mode_with_gpu.including("local_assert_no_cpu_op")
mode_local_assert = mode_local_assert.excluding("local_gpu_elemwise_1") old = config.assert_no_cpu_op
old = config.assert_no_cpu_op
#If the flag is raise # If the flag is raise
try: try:
config.assert_no_cpu_op = 'raise' config.assert_no_cpu_op = 'raise'
assert_raises(AssertionError, assert_raises(AssertionError, theano.function([], out,
theano.function([], out, mode=mode_local_assert))
mode=mode_local_assert)) finally:
finally: config.assert_no_cpu_op = old
#If the flag is ignore
config.assert_no_cpu_op = old # If the flag is ignore
try:
try: config.assert_no_cpu_op = 'ignore'
config.assert_no_cpu_op = 'ignore' theano.function([], out, mode=mode_local_assert)
fn = theano.function([], out, mode=mode_local_assert)) finally:
finally: config.assert_no_cpu_op = old
#If the flag is ignore
config.assert_no_cpu_op = old
def test_int_pow(): def test_int_pow():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论