提交 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):
utt.seed_rng()
self.mode = mode_with_gpu.excluding('constant_folding')
self.join_op = cuda.GpuJoin()
# No gpu split.
self.split_op_class = tensor.Split
self.split_op_class = cuda.GpuSplit
# No Make vector on the gpu, Join used instead
self.make_vector_op = cuda.GpuJoin()
self.floatX = "float32"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论