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