Commit Graph

4 Commits

Author SHA1 Message Date
Lillian Johnson
3b6f30824c OpInfo JIT op.output_func handling support (#50775)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50775

Test Plan: Imported from OSS

Reviewed By: mruberry

Differential Revision: D25964541

Pulled By: Lilyjjo

fbshipit-source-id: 8cf1ee9191d526cc46ae283f38c2d64bd60afdb2
2021-01-27 15:04:23 -08:00
Lillian Johnson
f1c578594b JIT Testing: Improve assertAutodiffNode error message (#50626)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50626

Test Plan: Imported from OSS

Reviewed By: mruberry

Differential Revision: D25932184

Pulled By: Lilyjjo

fbshipit-source-id: 6fa5a652eb1a0c10bb9d9040b9a708fdf93aaf46
2021-01-20 10:05:52 -08:00
Lillian Johnson
9c35a68094 Refactored assertAutodiff test to have better error message (#48567)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/48567

Test Plan: Imported from OSS

Reviewed By: eellison

Differential Revision: D25212435

Pulled By: Lilyjjo

fbshipit-source-id: eab3933bf4248dbfa20cd956d4a0106b10db5fc4
2020-12-02 19:59:33 -08:00
Lillian Johnson
c465602d78 Refactor existing JIT testing utils to enable new OpInfo test suite to reuse existing logic (#47695)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/47695

The method_tests from common_methods_invoations.py are being migrated into a new OpInfo class-based testing framework. The work in this commit pulls out the functions embedded in the old method_tests logic and places them in a location that both the old method_tests and OpInfo tests can use

Specifically: created torch/testing/_internal/common_jit.py from functions and methods in torch/testing/_internal/jit_utils.py and test/test_jit.py. Also created new intermediate class JitCommonTestCase to house moved methods. Also slightly modified jit_metaprogramming_utils.py to work for OpInfo tests

Test Plan: Imported from OSS

Reviewed By: mruberry

Differential Revision: D25212437

Pulled By: Lilyjjo

fbshipit-source-id: 97bc52c95d776d567750e7478fac722da30f4985
2020-12-02 19:54:30 -08:00