提交 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,32 +93,29 @@ def test_local_gpu_contiguous_gpu_contiguous(): ...@@ -92,32 +93,29 @@ 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_with_gpu.including("local_assert_no_cpu_op")
mode_local_assert = mode_local_assert.excluding("local_gpu_elemwise_1") 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:
#If the flag is ignore
config.assert_no_cpu_op = old 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'
fn = theano.function([], out, mode=mode_local_assert)) theano.function([], out, mode=mode_local_assert)
finally: finally:
#If the flag is ignore
config.assert_no_cpu_op = old config.assert_no_cpu_op = old
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论