提交 dbf22bbe authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Better fix the test.

上级 7399bc43
......@@ -3,7 +3,7 @@ import numpy
import theano
from theano import tensor
from theano.tests.breakpoint import PdbBreakpoint
from theano.tests import unittest_tools as utt
from theano.tests import unittest_tools as utt, test_ifelse
from theano.tensor.tests import test_basic
import theano.sandbox.gpuarray
......@@ -206,11 +206,12 @@ class TestSpecifyShape(test_basic.TestSpecifyShape):
input_type = GpuArrayType
class test_gpu_ifelse(theano.tests.test_ifelse.test_ifelse):
class test_gpu_ifelse(test_ifelse.test_ifelse):
mode = mode_with_gpu
@staticmethod
def cast_output(v):
return basic_ops.as_gpuarray_variable(v, test_ctx)
return basic_ops.as_gpuarray_variable(v, test_ctx_name)
shared = staticmethod(gpuarray_shared_constructor)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论