Updated kernel with flipped WG dimensions