pytorch
eb3e9872 - [JIT] make torch.unique compilable (#38156)

Commit
4 years ago
[JIT] make torch.unique compilable (#38156) Summary: Fix for https://github.com/pytorch/pytorch/issues/37986 Follows the stack in https://github.com/pytorch/pytorch/pull/33783 stack to make functions in `torch/functional.py` resolve to their python implementations. Because the return type of `torch.unique` depends on `return_inverse` and `return_counts` I had to refactor the implementation to use our boolean_dispatch mechanism. Pull Request resolved: https://github.com/pytorch/pytorch/pull/38156 Differential Revision: D21504449 Pulled By: eellison fbshipit-source-id: 7efb1dff3b5c00655da10168403ac4817286ff59
Author
Elias Ellison
Parents
Loading