pytorch

35ad5122 - Revert "[vmap] Fix searchsorted batch rule for self_logical_rank == 0 (#99526)"

Revert "[vmap] Fix searchsorted batch rule for self_logical_rank == 0 (#99526)" This reverts commit 6580b160d35a75d5ceebf376d55422376d0c0d2c. Reverted https://github.com/pytorch/pytorch/pull/99526 on behalf of https://github.com/zou3519 due to Regressed behavior