pytorch

274958ef - [vmap] unsafe_split : batching rule and OpInfo (#92291)

[vmap] unsafe_split : batching rule and OpInfo (#92291) Ref: https://github.com/pytorch/functorch/issues/1089 Pull Request resolved: https://github.com/pytorch/pytorch/pull/92291 Approved by: https://github.com/Chillee