提交 b12a695e authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #4127 from abergeron/fix_buildbot

Fix small buildbot problems
...@@ -47,7 +47,7 @@ class TestDnnConv2d(test_abstract_conv.BaseTestConv2d): ...@@ -47,7 +47,7 @@ class TestDnnConv2d(test_abstract_conv.BaseTestConv2d):
filter_flip=flip, target_op=GpuDnnConvGradI) filter_flip=flip, target_op=GpuDnnConvGradI)
class TestCorrMMConv2d(test_abstract_conv.TestConv2d): class TestCorrMMConv2d(test_abstract_conv.BaseTestConv2d):
def setUp(self): def setUp(self):
super(TestCorrMMConv2d, self).setUp() super(TestCorrMMConv2d, self).setUp()
self.shared = gpu_shared self.shared = gpu_shared
......
...@@ -1081,14 +1081,14 @@ def _as_scalar(res, dtype=None): ...@@ -1081,14 +1081,14 @@ def _as_scalar(res, dtype=None):
def _is_real_matrix(res): def _is_real_matrix(res):
return (res.type.dtype in ('float32', 'float64') and return (res.type.dtype in ('float32', 'float64') and
res.type.ndim == 2 and res.type.ndim == 2 and
res.type.broadcastable[0] == False and res.type.broadcastable[0] is False and
res.type.broadcastable[1] == False) # cope with tuple vs. list res.type.broadcastable[1] is False) # cope with tuple vs. list
def _is_real_vector(res): def _is_real_vector(res):
return (res.type.dtype in ('float32', 'float64') and return (res.type.dtype in ('float32', 'float64') and
res.type.ndim == 1 and res.type.ndim == 1 and
res.type.broadcastable[0] == False) res.type.broadcastable[0] is False)
def _beta_L_plus_alpha_M(beta, L, alpha, M, recurse_flip=True): def _beta_L_plus_alpha_M(beta, L, alpha, M, recurse_flip=True):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论