提交 5969400e authored 作者: Frederic's avatar Frederic

Enable the GpuSplit tests

上级 9b6f8491
...@@ -973,8 +973,7 @@ class T_Join_and_Split(theano.tensor.tests.test_basic.T_Join_and_Split): ...@@ -973,8 +973,7 @@ class T_Join_and_Split(theano.tensor.tests.test_basic.T_Join_and_Split):
utt.seed_rng() utt.seed_rng()
self.mode = mode_with_gpu.excluding('constant_folding') self.mode = mode_with_gpu.excluding('constant_folding')
self.join_op = cuda.GpuJoin() self.join_op = cuda.GpuJoin()
# No gpu split. self.split_op_class = cuda.GpuSplit
self.split_op_class = tensor.Split
# No Make vector on the gpu, Join used instead # No Make vector on the gpu, Join used instead
self.make_vector_op = cuda.GpuJoin() self.make_vector_op = cuda.GpuJoin()
self.floatX = "float32" self.floatX = "float32"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论