pytorch
30558c28 - [functorch] Get test_functionalize to run on FB infra (#102695)

Commit
1 year ago
[functorch] Get test_functionalize to run on FB infra (#102695) A few bits of weirdness needed to happen here: - skipIfRocm doesn't work as a unittest class decorator; it returns a function, and the test discovery logic looks for things that inherit from TestCase. So I wrapped the individual test methods instead. - Inside fbcode, our test runner (buck + tpx) discovers and runs tests using two separate processes, so it's important to use @wraps on the generated class to make it "look like" a regular test. Differential Revision: [D46344980](https://our.internmc.facebook.com/intern/diff/D46344980/) **NOTE FOR REVIEWERS**: This PR has internal Meta-specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D46344980/)! Pull Request resolved: https://github.com/pytorch/pytorch/pull/102695 Approved by: https://github.com/zou3519
Author
Committer
Parents
Loading