k_blockreduce computes gradient too