Merge "[PiP2] Cache task info in PipTransitionState" into main