kernel里面能并行numpy array吗?

今天才发现,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
...

看样子是并行的?我不确定,希望有清楚的大佬告知一下,谢谢!

是并行的。