提交 cbceb887 authored 作者: Frederic's avatar Frederic

make test execute more often on the GPU.

上级 de717450
...@@ -888,6 +888,8 @@ class T_subtensor(theano.tensor.tests.test_basic.T_subtensor): ...@@ -888,6 +888,8 @@ class T_subtensor(theano.tensor.tests.test_basic.T_subtensor):
dtype = 'float32' dtype = 'float32'
ignore_topo = (B.HostFromGpu, B.GpuFromHost) ignore_topo = (B.HostFromGpu, B.GpuFromHost)
fast_compile = theano.config.mode == 'FAST_COMPILE' fast_compile = theano.config.mode == 'FAST_COMPILE'
ops = (cuda.GpuSubtensor, cuda.GpuIncSubtensor,
cuda.GpuAdvancedSubtensor1, cuda.GpuAdvancedIncSubtensor1)
def __init__(self, name): def __init__(self, name):
return super(theano.tensor.tests.test_basic.T_subtensor, return super(theano.tensor.tests.test_basic.T_subtensor,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论