Different TB_SIZE in 03_sum_reduction/diverged won't pass assertion

I found that a TB_SIZE and SH_SIZE= 128, won't give the expected 65536 result.