Taichi支持枚举类型么?

综合讨论区
#1

如题所示

#2

用i32表示
RED = 1
GREEN = 2
BLUE = 3
就行了

#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