提交 39a74766 authored 作者: AdeB's avatar AdeB 提交者: Pascal Lamblin

Remove the ignore flag and the check_trace where the ops were missing

上级 baddb1d6
......@@ -77,8 +77,7 @@ def pyconv3d(signals, filters):
def check_diagonal_subtensor_view_traces(fn):
assert check_stack_trace(
fn, ops_to_check=(DiagonalSubtensor, IncDiagonalSubtensor),
bug_print='ignore')
fn, ops_to_check=(DiagonalSubtensor, IncDiagonalSubtensor))
def test_conv3d(mode=mode_without_gpu, shared=theano.tensor._shared):
......@@ -153,7 +152,6 @@ def test_conv3d(mode=mode_without_gpu, shared=theano.tensor._shared):
newconv3d = theano.function([], [],
updates={s_output: out},
mode=mode)
check_diagonal_subtensor_view_traces(newconv3d)
t0 = time.time()
newconv3d()
......@@ -165,7 +163,6 @@ def test_conv3d(mode=mode_without_gpu, shared=theano.tensor._shared):
(s_signals, gsignals)],
mode=mode,
name='grad')
check_diagonal_subtensor_view_traces(gnewconv3d)
t0 = time.time()
gnewconv3d()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论