提交 bfd25c47 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Don't try to test the lift when it's not available.

上级 6efc48cd
......@@ -17,9 +17,9 @@ from ..basic_ops import (
from ..blas import GpuGemm
from ..elemwise import GpuCAReduceCuda, GpuCAReduceCPY, GpuElemwise
from ..subtensor import GpuSubtensor
from ..linalg import GpuCusolverSolve
from ..linalg import GpuCusolverSolve, cusolver_available
from .config import mode_with_gpu, test_ctx_name
from .config import mode_with_gpu, test_ctx_name, SkipTest
def test_local_assert():
......@@ -501,6 +501,8 @@ def test_no_complex():
def test_local_lift_solve():
if not cusolver_available:
raise SkipTest('No cuSolver')
A = tensor.fmatrix()
b = tensor.fmatrix()
o = slinalg.solve(A, b)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论