提交 52413fc6 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Remove testing code from ScanSaveMem

上级 bf5938c7
......@@ -1569,8 +1569,6 @@ class ScanSaveMem(GlobalOptimizer):
# 3.6 Compose the new scan
# TODO: currently we don't support scan with 0 step. So
# don't create one.
# For test, mark that savemem have optimized this node
info["_scan_savemem_visited"] = True
if aet.extract_constant(node_ins[0]) == 0:
return
......
......@@ -371,11 +371,13 @@ class TestScan:
aesara_values = my_f(state, steps)
utt.assert_allclose(numpy_values, aesara_values[0])
@pytest.mark.xfail(
reason="This is a very poorly specified test that needs to be rewritten"
)
def test_subtensor_multiple_slices(self):
# This addresses a bug reported by Matthias Zoehrer
# the bug happens when you have multiple subtensors on the output of
# scan (the bug requires the reshape to be produced, and it has
# which has something to do with how the subtensors overlap
# This addresses a bug that happens when you have multiple subtensors
# on the output of scan (the bug requires the reshape to be produced,
# and it has which has something to do with how the subtensors overlap
def f_pow2(x_tm1):
return 2 * x_tm1
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论