1. set xpu2 block_size = 64
2. fix a bug when reduce_num is too large