请问把代码提炼到ti.func之后最外层for还能并行吗

综合讨论区
#1

请问把代码提炼到ti.func之后最外层for还能并行吗

例如

原本

@ti.kernel
def func_outer():
    for i in xxx:
        ...

现在把里面的for循环提炼出来，然后用func_outer调用

@ti.func
def func_inner():
    for i in xxx:
        ...

@ti.kernel
def func_outer():
    func_inner()

我的本意是重构并且提炼代码。在我印象中ti.func只是单纯的把代码复制粘贴到原来的位置。但是我不确定现在还是否能够并行

#2

Hi @beidou , 在 ti.func 里的最外层for循环也是可以并行的。