提交 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)}):
{indent(input_storage_block, " " * 4)}
i = 0
cond = False
while i < n_steps and not cond:
cond = np.array(False)
while i < n_steps and not cond.item():
{inner_outputs} = scan_inner_func({inner_in_args})
{indent(inner_out_post_processing_block, " " * 8)}
{indent(inner_out_to_outer_out_stmts, " " * 8)}
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论