Test cases for custom ops with autograd (#31003)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/31003
-
ghstack-source-id: 95663728
Test Plan: unit tests
Differential Revision: D18896189
fbshipit-source-id: d71f7678fff644536fe30452ee21a5a7df1f1f0b