提交 fb964cdb authored 作者: Frederic Bastien's avatar Frederic Bastien

don't bypass DebugMode.

上级 611a3d75
...@@ -1286,7 +1286,7 @@ class TestOptimizationMixin(object): ...@@ -1286,7 +1286,7 @@ class TestOptimizationMixin(object):
class TestGer_local_gemm_to_ger(TestCase, TestOptimizationMixin): class TestGer_local_gemm_to_ger(TestCase, TestOptimizationMixin):
def setUp(self): def setUp(self):
self.mode = theano.Mode(optimizer='fast_run') self.mode = theano.compile.get_default_mode().including('fast_run')
dtype = self.dtype = 'float64' # optimization isn't dtype-dependent dtype = self.dtype = 'float64' # optimization isn't dtype-dependent
self.A = T.tensor(dtype=dtype, broadcastable=(False, False)) self.A = T.tensor(dtype=dtype, broadcastable=(False, False))
self.a = T.tensor(dtype=dtype, broadcastable=()) self.a = T.tensor(dtype=dtype, broadcastable=())
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论