Modify the warp-probe kernel. Now we can set the number of warps per block.