运行环境
[Taichi] mode=release
[Taichi] preparing sandbox at /tmp/taichi-ppid_map
[Taichi] version 0.7.29, llvm 10.0.0, commit 7b40a962, linux, python 3.7.10
[I 10/13/21 15:21:05.818 28275] [shell.py:_shell_pop_print@35] Graphical python shell detected, using wrapped sys.stdout
[Taichi] Starting on arch=cuda
问题描述
我在用jupyter练习基本语法是发现了一个比较奇怪的错误，希望有人可以解释一下～
具体来说，当我把_add里的注释去掉，会报错。然后我再把这一行注释，就会报kernel不能嵌套的错误。遇到kernal不能嵌套这个错误这让我比较疑惑，因为我只写了一个kernel呀。
import taichi as ti
ti.init(arch=ti.gpu, default_fp=ti.f32)
@ti.func
def _add(x,y):
res = 0
if x>1:
res = x+y
# return res
else:
res = x-y
return res
@ti.kernel
def demo(x:ti.i32, y:ti.i32) -> ti.i32:
z = _add(x,y)
return z
print(demo(1,2))
