提交 63006f46 authored 作者: Frédéric Bastien's avatar Frédéric Bastien 提交者: GitHub

Merge pull request #5390 from abergeron/fix_jenkins

Fix the buildbot.
...@@ -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)
......
...@@ -1023,9 +1023,9 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -1023,9 +1023,9 @@ class TestDnnInferShapes(utt.InferShapeTester):
@parameterized.expand(product(border_modes, conv_modes), utt.custom_name_func) @parameterized.expand(product(border_modes, conv_modes), utt.custom_name_func)
def test_conv_gradw(self, border_mode, conv_mode): def test_conv_gradw(self, border_mode, conv_mode):
self._test_conv_gradw(T.tensor4('img'), self._test_conv_gradw(T.ftensor4('img'),
T.tensor4('topgrad'), T.ftensor4('topgrad'),
T.tensor4('kerns'), T.ftensor4('kerns'),
(5, 2, 6, 13), (5, 2, 6, 13),
(1, 2, 3, 7), (1, 2, 3, 7),
border_mode, border_mode,
...@@ -1071,9 +1071,9 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -1071,9 +1071,9 @@ class TestDnnInferShapes(utt.InferShapeTester):
@parameterized.expand(product(border_modes, conv_modes), utt.custom_name_func) @parameterized.expand(product(border_modes, conv_modes), utt.custom_name_func)
def test_conv3d_gradw(self, border_mode, conv_mode): def test_conv3d_gradw(self, border_mode, conv_mode):
self._test_conv3d_gradw(T.tensor5('img'), self._test_conv3d_gradw(T.ftensor5('img'),
T.tensor5('topgrad'), T.ftensor5('topgrad'),
T.tensor5('kerns'), T.ftensor5('kerns'),
(5, 2, 6, 13, 21), (5, 2, 6, 13, 21),
(1, 2, 3, 7, 9), (1, 2, 3, 7, 9),
border_mode, border_mode,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论