提交 53abe8ea authored 作者: Frederic Bastien's avatar Frederic Bastien

fix test on the gpu.

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