提交 67e14ae0 authored 作者: Frederic Bastien's avatar Frederic Bastien

Pre apply opt.

上级 644d8bce
...@@ -2724,6 +2724,7 @@ def merge_two_slices(slice1, len1, slice2, len2): ...@@ -2724,6 +2724,7 @@ def merge_two_slices(slice1, len1, slice2, len2):
val = T.switch(T.lt(sl2, 0), - len1 - 1, val) val = T.switch(T.lt(sl2, 0), - len1 - 1, val)
if sl1.step: if sl1.step:
val = T.switch(T.eq(sl1.step, 0), len1 + 1, val) val = T.switch(T.eq(sl1.step, 0), len1 + 1, val)
val = pre_greedy_local_optimizer(list_opt, val)
return val return val
else: else:
# We are in the more complex case when we do not actually know # We are in the more complex case when we do not actually know
...@@ -2748,6 +2749,7 @@ def merge_two_slices(slice1, len1, slice2, len2): ...@@ -2748,6 +2749,7 @@ def merge_two_slices(slice1, len1, slice2, len2):
val = T.switch(T.lt(sl2, 0), - len1 - 1, val) val = T.switch(T.lt(sl2, 0), - len1 - 1, val)
if sl1.step: if sl1.step:
val = T.switch(T.eq(sl1.step, 0), len1 + 1, val) val = T.switch(T.eq(sl1.step, 0), len1 + 1, val)
val = pre_greedy_local_optimizer(list_opt, val)
return val return val
else: else:
# We are deleaing with two slices that need to be put together # We are deleaing with two slices that need to be put together
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论