pytorch
13011afb - Fix vmap registration for t, t_ (#96539)

Commit
2 years ago
Fix vmap registration for t, t_ (#96539) - t, t_ are not CompositeImplicitAutograd - They were previously registered in BatchRulesDecompositions.cpp. - The only thing that should get registered in BatchRulesDecompositions.cpp are CompositeImplicitAutograd - This PR moves their registrations out of there and into BatchRulesViews.cpp. Test Plan: - existing tests Pull Request resolved: https://github.com/pytorch/pytorch/pull/96539 Approved by: https://github.com/srossross, https://github.com/kshitij12345, https://github.com/Chillee
Author
Committer
Parents
Loading