[TensorExpr] Add helper passes for AOT pipeline. (#66514)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/66514
These passes will
1) help us analyze the graph before trying to compile it
and report errors upfront if it's not possible,
2) fill in missing strides/dtype/device info in JIT IR. Ideally, this
should be done by a dedicated JIT pass, but until it's available, we'll
be using a hack-around defined here.
Differential Revision:
D31590860
D31590860
Test Plan: Imported from OSS
Reviewed By: bdhirsh
Pulled By: ZolotukhinM
fbshipit-source-id: fe8fdefbeacae8079958dd0b4b27809cc0acb34b
Author
Mikhail Zolotukhin