tree 609b7945a1e802d2c1ef06c5f13548aa1b621a14
parent 71510c60ada8b4a13e9d5585cf47fd9054c1ee7a
author Mikhail Zolotukhin <mvz@fb.com> 1599175483 -0700
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> 1599175634 -0700

[TensorExpr] Fuser: rely on input types when checking whether a device is supported. (#44139)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/44139

Also, make sure that we're checking that condition when we're starting a
new fusion group, not only when we merge a node into an existing fusion
group. Oh, and one more: add a test checking that we're rejecting graphs
with unspecified shapes.

Differential Revision: D23507510

Test Plan: Imported from OSS

Reviewed By: bertmaher

Pulled By: ZolotukhinM

fbshipit-source-id: 9c268825ac785671d7c90faf2aff2a3e5985ac5b