今天才发现,numpy array作为参数也是能够传入kernel并且被循环的。
但是我不清楚这个np arr是不是并行的
于是打印了一下0到99
import taichi as ti
ti.init(ti.gpu)
a = ti.ndarray(dtype=ti.i32, shape=(100))
@ti.kernel
def test(a: ti.types.ndarray()):
for i in range(100):
a[i] = i
print(a[i])
test(a)
output:
96
97
98
99
0
1
2
3
4
5
6
7
8
9
...
看样子是并行的?我不确定,希望有清楚的大佬告知一下,谢谢!