taichi gpu的gridDim与blockDim

综合讨论区
#1

请问taichi默认的gridDim与blockDim是多少呢，应该通过什么函数来获取
如何获取blockId.x/y/z
以及如何修改gridDim

#2 
@ti.kernel
def func():
    for i in x:
        ti.loop_config(block_dim=42)
        x[i] = i
#3

这不对把，这只是设置blockDim，和我的问题好像关系不大

#4

没必要获取blockdim，太极都是for循环，cpu核gpu统一的，用不着获取idx然后乘，for i in range(n)直接i就是最后的索引。

#5

因为我想做reduce max，我看taichi里的ti.simt.subgroup.reduce_max好像还不能用

#6

直接atomic_max就行，太极是傻瓜式的，会自动把所有对单个变量的atomic“优化”成BLS的，不需要用户考虑reduce的具体实现细节。

x = ti.field(shape=1024, dt=ti.f32)
res = ti.field(shape=(), dt=ti.f32)

@ti.kernel
def findmax():
  for i in x:
    ti.atomic_max(res[None], x[i])

findmax()
print(res[None])