怎样手写一个并行的field求和kernel函数

我想对一个field的数据求和。一般的并行算法是并行计算i与i+2**n位置的和,n从logN一直递减到0。请问这个过程怎么写成taichi的kernel函数?