pytorch
3ec24f02 - [TensorExpr] Start using typecheck in the fuser. (#43173)

Commit
5 years ago
[TensorExpr] Start using typecheck in the fuser. (#43173) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/43173 With this change the fuser starts to generate typechecks for inputs of fusion group. For each fusion group we generate a typecheck and an if node: the true block contains the fused subgraph, the false block contains unoptimized original subgraph. Differential Revision: D23178230 Test Plan: Imported from OSS Reviewed By: eellison Pulled By: ZolotukhinM fbshipit-source-id: f56e9529613263fb3e6575869fdb49973c7a520b
Author
Mikhail Zolotukhin
Parents
Loading