提交 6980abc7 authored 作者: Frederic's avatar Frederic 提交者: Arnaud Bergeron

Make the new GpuSubtensor tests run with it!

上级 8f21a69a
......@@ -7,6 +7,8 @@ from theano.sandbox.gpuarray.type import gpuarray_shared_constructor
from theano.sandbox.gpuarray.tests.test_basic_ops import mode_with_gpu
from theano import tensor
class G_subtensor(T_subtensor):
def shortDescription(self):
return None
......@@ -17,7 +19,19 @@ class G_subtensor(T_subtensor):
dtype = 'float32' # avoid errors on gpus which do not support float64
ignore_topo = (HostFromGpu, GpuFromHost)
fast_compile = False
ops = (GpuSubtensor,)
ops = (GpuSubtensor,
tensor.IncSubtensor, tensor.AdvancedSubtensor1,
tensor.AdvancedIncSubtensor1)
#inc_sub = cuda.GpuIncSubtensor
#adv_sub1 = cuda.GpuAdvancedSubtensor1
#adv_incsub1 = cuda.GpuAdvancedIncSubtensor1
inc_sub = tensor.IncSubtensor
adv_sub1 = tensor.AdvancedSubtensor1
adv_incsub1 = tensor.AdvancedIncSubtensor1
def __init__(self, name):
T_subtensor.__init__(self, name)
# We must call T_subtensor parten __init__, otherwise, sub is overrieded
ret = super(T_subtensor, self).__init__(name)
assert self.sub == GpuSubtensor
return ret
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论