提交 0c2a2b08 authored 作者: Adrian Seyboldt's avatar Adrian Seyboldt 提交者: Adrian Seyboldt

numba while condition must be tensor

上级 33769035
...@@ -351,8 +351,8 @@ def scan({", ".join(outer_in_names)}): ...@@ -351,8 +351,8 @@ def scan({", ".join(outer_in_names)}):
{indent(input_storage_block, " " * 4)} {indent(input_storage_block, " " * 4)}
i = 0 i = 0
cond = False cond = np.array(False)
while i < n_steps and not cond: while i < n_steps and not cond.item():
{inner_outputs} = scan_inner_func({inner_in_args}) {inner_outputs} = scan_inner_func({inner_in_args})
{indent(inner_out_post_processing_block, " " * 8)} {indent(inner_out_post_processing_block, " " * 8)}
{indent(inner_out_to_outer_out_stmts, " " * 8)} {indent(inner_out_to_outer_out_stmts, " " * 8)}
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论