提交 24b60c57 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

We need to ignore the new GpuContiguous op.

上级 43eec188
...@@ -7,7 +7,7 @@ from theano import tensor ...@@ -7,7 +7,7 @@ from theano import tensor
from theano.compile import DeepCopyOp from theano.compile import DeepCopyOp
from theano.tensor.tests import test_subtensor from theano.tensor.tests import test_subtensor
from ..basic_ops import HostFromGpu, GpuFromHost from ..basic_ops import HostFromGpu, GpuFromHost, GpuContiguous
from ..elemwise import GpuDimShuffle from ..elemwise import GpuDimShuffle
from ..subtensor import (GpuIncSubtensor, GpuSubtensor, from ..subtensor import (GpuIncSubtensor, GpuSubtensor,
GpuAdvancedSubtensor1, GpuAdvancedSubtensor1,
...@@ -37,7 +37,7 @@ class G_subtensor(test_subtensor.T_subtensor): ...@@ -37,7 +37,7 @@ class G_subtensor(test_subtensor.T_subtensor):
# avoid errors with limited devices # avoid errors with limited devices
dtype='float32', dtype='float32',
ignore_topo=(HostFromGpu, GpuFromHost, ignore_topo=(HostFromGpu, GpuFromHost,
DeepCopyOp)) DeepCopyOp, GpuContiguous))
# GPU opt can't run in fast_compile only. # GPU opt can't run in fast_compile only.
self.fast_compile = False self.fast_compile = False
assert self.sub == GpuSubtensor assert self.sub == GpuSubtensor
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论