pytorch
98b79e94 - [inductor] Add AOTI ABI shim function for torch.nonzero (#110766)

Commit
1 year ago
[inductor] Add AOTI ABI shim function for torch.nonzero (#110766) Summary: `torch.nonzero` doesn't have inductor lowering (yet). To invoke the operator in AOT Inductor's ABI compatibility mode we need a dedicated shim function. Test Plan: ``` $ python test/inductor/test_aot_inductor.py -k test_zero_grid_with_unbacked_symbols ... ---------------------------------------------------------------------- Ran 4 tests in 78.650s OK ``` Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/110766 Approved by: https://github.com/chenyang78 ghstack dependencies: #110713, #110745, #110764
Author
Committer
Parents
Loading