pytorch
5a82770a - add ssupport for a few more opinfos in jit (#59812)

Commit
3 years ago
add ssupport for a few more opinfos in jit (#59812) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/59812 This is sort of a half measure: we can successfully trace through opinfos which are registered as lambdas, we just can't script them. This tests if the op is a lambda in which case bails... see the next PR to get resize_ to work, maybe this should be consolidated with that... Test Plan: Imported from OSS Reviewed By: pbelevich, zhxchen17 Differential Revision: D30200061 Pulled By: eellison fbshipit-source-id: 7e3c9b0be746b16f0f57ece49f6fbe20bf6535ec
Author
Elias Ellison
Committer
Parents
Loading