Each thread block now handles 4 rows/columns