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