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
