def f(arr): x, y, z = jit.grid(3) gx, gy, gz = jit.gridsize(3) if x == 0 and y == 0 and z == 0: arr[0] = gx arr[1] = gy arr[2] = gz
def f(arr): x, y = jit.grid(2) gx, gy = jit.gridsize(2) if x == 0 and y == 0: arr[0] = gx arr[1] = gy
def f(arr): x = jit.grid(1) gx = jit.gridsize(1) if x == 0: arr[x] = gx