如题所示
用i32表示
RED = 1
GREEN = 2
BLUE = 3
就行了
用python的枚举
import enum
import taichi as ti
ti.init(arch = ti.cpu)
class COLOR(enum.IntEnum):
RED = enum.auto()
GREEN = enum.auto()
BLUE = enum.auto()
@ti.kernel
def f():
c = COLOR.RED
print(c)
c = COLOR.GREEN
print(c)
f()
1 Like