提交 c62dec67 authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Skip GpuSolve test if CULA is not available

上级 576c2624
...@@ -19,6 +19,8 @@ import theano.sandbox.cuda as cuda ...@@ -19,6 +19,8 @@ import theano.sandbox.cuda as cuda
if not cuda.cuda_available: if not cuda.cuda_available:
raise SkipTest('Optional package cuda disabled') raise SkipTest('Optional package cuda disabled')
import theano.sandbox.cuda.cula as cula
from theano.sandbox.cuda import basic_ops from theano.sandbox.cuda import basic_ops
from theano.sandbox.cuda.type import CudaNdarrayType from theano.sandbox.cuda.type import CudaNdarrayType
from theano.scalar.basic_scipy import erfinv from theano.scalar.basic_scipy import erfinv
...@@ -537,6 +539,10 @@ def test_erfinvgpu(): ...@@ -537,6 +539,10 @@ def test_erfinvgpu():
def test_local_gpu_solve(): def test_local_gpu_solve():
if not cula.cula_available:
raise SkipTest('Optional dependency CULA not available')
numpy.random.seed(1) numpy.random.seed(1)
def cmp(a_shp, b_shp): def cmp(a_shp, b_shp):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论