Set Z threads per block to zero when 2d + axis chosen