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

Refactor test to have more tests run on the GPU.

上级 91739a95
......@@ -2,7 +2,7 @@ from theano import scalar, gof
from theano.gof.python25 import all, any
from theano.tensor.tests.test_elemwise import (test_Broadcast, test_DimShuffle,
test_CAReduce, T_sum_dtype)
test_CAReduce, T_reduce_dtype)
from theano.sandbox.gpuarray.tests.test_basic_ops import (mode_with_gpu,
rand_gpuarray)
......@@ -152,7 +152,7 @@ class test_GpuCAReduceCuda(test_GpuCAReduceCPY):
return
class T_gpusum_dtype(T_sum_dtype):
class T_gpureduce_dtype(T_reduce_dtype):
mode = mode_with_gpu.excluding('local_cut_useless_reduce')
op = GpuCAReduceCuda
#Currently we don't support reduction on 0 axis
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论