提交 76357631 authored 作者: Frederic Bastien's avatar Frederic Bastien

'run all subtensor test in tensor/tests/test_basic.py also for gpu subtensor.'

上级 c5727d8c
...@@ -785,10 +785,11 @@ def test_gpualloc_output_to_gpu(): ...@@ -785,10 +785,11 @@ def test_gpualloc_output_to_gpu():
import theano.tensor.tests.test_basic import theano.tensor.tests.test_basic
# This is to don't duplicate test. # This is to don't duplicate test.
# TODO: the source class test only Adv_subtensor1 test on gpu. All other are tested only on the cpu! class T_subtensor(theano.tensor.tests.test_basic.T_subtensor):
class T_Adv_subtensor1(theano.tensor.tests.test_basic.T_subtensor):
shared=staticmethod(cuda.shared_constructor) shared=staticmethod(cuda.shared_constructor)
adv_sub1=cuda.GpuAdvancedSubtensor1 adv_sub1=cuda.GpuAdvancedSubtensor1
sub=cuda.GpuSubtensor,
inc_sub=cuda.GpuIncSubtensor,
mode=mode_with_gpu mode=mode_with_gpu
dtype='float32' dtype='float32'
ignore_topo=(B.HostFromGpu, B.GpuFromHost) ignore_topo=(B.HostFromGpu, B.GpuFromHost)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论