提交 3f5b76b1 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Thomas Wiecki

Numba scan: only rotate outputs if indexing wraps around storage size

上级 e5bdedb8
......@@ -190,8 +190,8 @@ def numba_funcify_Scan(op, node, **kwargs):
output_storage_post_proc_stmts.append(
dedent(
f"""
if (i + {tap_size}) > {storage_size}:
{outer_in_name}_shift = (i + {tap_size}) % ({storage_size})
if {outer_in_name}_shift > 0:
{outer_in_name}_left = {outer_in_name}[:{outer_in_name}_shift]
{outer_in_name}_right = {outer_in_name}[{outer_in_name}_shift:]
{outer_in_name} = np.concatenate(({outer_in_name}_right, {outer_in_name}_left))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论