Setup the correction_rescale atoms BEFORE the softmax loop so they can be shared between per-tile O rescale and final normalize. Uses the working 2D register tensor pattern for final normalize. O rescale uses simple 1D rmem tensor per sub-tile (same as example10).